test conversion: move old-style test to new-style test

astraw · astraw · commit f3144f3f8619 · 2009-09-07T20:05:12.000Z
svn path=/trunk/matplotlib/; revision=7694
diff --git a/lib/matplotlib/tests/baseline_images/test_axes/axvspan_epoch.png b/lib/matplotlib/tests/baseline_images/test_axes/axvspan_epoch.png
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -312,3 +312,25 @@ def test_polar_units():
     # polar( x2, y2, color = "green" )
 
     fig.savefig( 'polar_units' )
+
+@image_comparison(baseline_images=['axvspan_epoch'])
+def test_axvspan_epoch():
+    """Test the axvspan method with Epochs."""
+    from datetime import datetime
+    import matplotlib.testing.jpl_units as units
+    units.register()
+
+    # generate some data
+    t0 = units.Epoch( "ET", dt=datetime(2009, 1, 20) )
+    tf = units.Epoch( "ET", dt=datetime(2009, 1, 21) )
+
+    dt = units.Duration( "ET", units.day.convert( "sec" ) )
+
+    fig = pylab.figure()
+
+    pylab.axvspan( t0, tf, facecolor="blue", alpha=0.25 )
+
+    ax = pylab.gca()
+    ax.set_xlim( t0 - 5.0*dt, tf + 5.0*dt )
+
+    fig.savefig( 'axvspan_epoch' )
diff --git a/test/test_plots/TestSpan.py b/test/test_plots/TestSpan.py
@@ -41,28 +41,6 @@ def tearDown( self ):
       """Clean-up any generated files here."""
       pass
 
-   #--------------------------------------------------------------------
-   def test_axvspan_epoch( self ):
-      """Test the axvspan method with Epochs."""
-
-      fname = self.outFile( "axvspan_epoch.png" )
-
-      # generate some data
-      t0 = units.Epoch( "ET", dt=datetime(2009, 1, 20) )
-      tf = units.Epoch( "ET", dt=datetime(2009, 1, 21) )
-
-      dt = units.Duration( "ET", units.day.convert( "sec" ) )
-
-      fig = pylab.figure()
-
-      pylab.axvspan( t0, tf, facecolor="blue", alpha=0.25 )
-
-      ax = pylab.gca()
-      ax.set_xlim( t0 - 5.0*dt, tf + 5.0*dt )
-
-      fig.savefig( fname )
-      self.checkImage( fname )
-
    #--------------------------------------------------------------------
    def test_axhspan_epoch( self ):
       """Test the axhspan method with Epochs."""
