Fix test

mdboom · mdboom · commit 6deb32363e9d · 2014-12-03T11:37:48.000-05:00
diff --git a/lib/matplotlib/tests/test_agg.py b/lib/matplotlib/tests/test_agg.py
@@ -134,15 +134,13 @@ def report_memory(i):
 def test_marker_with_nan():
     # This creates a marker with nans in it, which was segfaulting the
     # Agg backend (see #3722)
-    fig = Figure()
+    fig, ax = plt.subplots(1)
     steps = 1000
     data = np.arange(steps)
-    ax = fig.add_subplot(1, 1, 1)
     ax.semilogx(data)
     ax.fill_between(data, data*0.8, data*1.2)
     buf = io.BytesIO()
-    fig.savefig(buf)
-
+    fig.savefig(buf, format='png')
 
 
 if __name__ == "__main__":
