Fixed tests to include svg and pdf for figimage

jklymak · jklymak · commit 272b377a21cd · 2017-09-19T22:05:38.000-07:00
diff --git a/lib/matplotlib/tests/test_image.py b/lib/matplotlib/tests/test_image.py
@@ -72,11 +72,12 @@ def test_interp_nearest_vs_none():
     ax2.set_title('interpolation nearest')
 
 
-@image_comparison(baseline_images=['figimage-0', 'figimage-1'], extensions=['png'])
+@image_comparison(baseline_images=['figimage-0'],
+                  extensions=['png','svg','pdf'])
 def test_figimage():
     'test the figimage method'
 
-    for suppressComposite in False, True:
+    for suppressComposite in [False]:
         fig = plt.figure(figsize=(2,2), dpi=100)
         fig.suppressComposite = suppressComposite
         x,y = np.ix_(np.arange(100.0)/100.0, np.arange(100.0)/100.0)
@@ -89,6 +90,25 @@ def test_figimage():
         fig.figimage(img[:,::-1], xo=100, yo=0, origin='lower')
         fig.figimage(img[::-1,::-1], xo=100, yo=100, origin='lower')
 
+@image_comparison(baseline_images=['figimage-1'],
+                  extensions=['png','svg','pdf'])
+def test_figimage():
+    'test the figimage method'
+
+    for suppressComposite in [True]:
+        fig = plt.figure(figsize=(2,2), dpi=100)
+        fig.suppressComposite = suppressComposite
+        x,y = np.ix_(np.arange(100.0)/100.0, np.arange(100.0)/100.0)
+        z = np.sin(x**2 + y**2 - x*y)
+        c = np.sin(20*x**2 + 50*y**2)
+        img = z + c/5
+
+        fig.figimage(img, xo=0, yo=0, origin='lower')
+        fig.figimage(img[::-1,:], xo=0, yo=100, origin='lower')
+        fig.figimage(img[:,::-1], xo=100, yo=0, origin='lower')
+        fig.figimage(img[::-1,::-1], xo=100, yo=100, origin='lower')
+
+
 def test_image_python_io():
     fig = plt.figure()
     ax = fig.add_subplot(111)
