Catch warnings in tightlayout test.

jenshnielsen · jenshnielsen · commit 155e1f7a6c61 · 2014-07-26T11:08:59.000+02:00
These are warnings that are raised because tightlayout does not work
fully automatic on gridspec. However, the point of the test is doing
this with some manual help so this works as expected.
diff --git a/lib/matplotlib/tests/test_tightlayout.py b/lib/matplotlib/tests/test_tightlayout.py
@@ -2,6 +2,7 @@
                         unicode_literals)
 
 import six
+import warnings
 
 import numpy as np
 
@@ -10,12 +11,14 @@
 from nose.tools import assert_raises
 from numpy.testing import assert_array_equal
 
+
 def example_plot(ax, fontsize=12):
-     ax.plot([1, 2])
-     ax.locator_params(nbins=3)
-     ax.set_xlabel('x-label', fontsize=fontsize)
-     ax.set_ylabel('y-label', fontsize=fontsize)
-     ax.set_title('Title', fontsize=fontsize)
+    ax.plot([1, 2])
+    ax.locator_params(nbins=3)
+    ax.set_xlabel('x-label', fontsize=fontsize)
+    ax.set_ylabel('y-label', fontsize=fontsize)
+    ax.set_title('Title', fontsize=fontsize)
+
 
 @image_comparison(baseline_images=['tight_layout1'])
 def test_tight_layout1():
@@ -81,50 +84,51 @@ def test_tight_layout5():
     fig = plt.figure()
 
     ax = plt.subplot(111)
-    arr = np.arange(100).reshape((10,10))
+    arr = np.arange(100).reshape((10, 10))
     ax.imshow(arr, interpolation="none")
 
     plt.tight_layout()
 
 
-
 @image_comparison(baseline_images=['tight_layout6'])
 def test_tight_layout6():
     'Test tight_layout for gridspec'
 
-    fig = plt.figure()
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore", UserWarning)
+        fig = plt.figure()
 
-    import matplotlib.gridspec as gridspec
+        import matplotlib.gridspec as gridspec
 
-    gs1 = gridspec.GridSpec(2, 1)
-    ax1 = fig.add_subplot(gs1[0])
-    ax2 = fig.add_subplot(gs1[1])
+        gs1 = gridspec.GridSpec(2, 1)
+        ax1 = fig.add_subplot(gs1[0])
+        ax2 = fig.add_subplot(gs1[1])
 
-    example_plot(ax1)
-    example_plot(ax2)
+        example_plot(ax1)
+        example_plot(ax2)
 
-    gs1.tight_layout(fig, rect=[0, 0, 0.5, 1])
+        gs1.tight_layout(fig, rect=[0, 0, 0.5, 1])
 
-    gs2 = gridspec.GridSpec(3, 1)
+        gs2 = gridspec.GridSpec(3, 1)
 
-    for ss in gs2:
-        ax = fig.add_subplot(ss)
-        example_plot(ax)
-        ax.set_title("")
-        ax.set_xlabel("")
+        for ss in gs2:
+            ax = fig.add_subplot(ss)
+            example_plot(ax)
+            ax.set_title("")
+            ax.set_xlabel("")
 
-    ax.set_xlabel("x-label", fontsize=12)
+        ax.set_xlabel("x-label", fontsize=12)
 
-    gs2.tight_layout(fig, rect=[0.5, 0, 1, 1], h_pad=0.45)
+        gs2.tight_layout(fig, rect=[0.5, 0, 1, 1], h_pad=0.45)
 
-    top = min(gs1.top, gs2.top)
-    bottom = max(gs1.bottom, gs2.bottom)
+        top = min(gs1.top, gs2.top)
+        bottom = max(gs1.bottom, gs2.bottom)
 
-    gs1.tight_layout(fig, rect=[None, 0 + (bottom-gs1.bottom),
-                                0.5, 1 - (gs1.top-top)])
-    gs2.tight_layout(fig, rect=[0.5, 0 + (bottom-gs2.bottom),
-                                None, 1 - (gs2.top-top)],
-                     h_pad=0.45)
+        gs1.tight_layout(fig, rect=[None, 0 + (bottom-gs1.bottom),
+                                    0.5, 1 - (gs1.top-top)])
+        gs2.tight_layout(fig, rect=[0.5, 0 + (bottom-gs2.bottom),
+                                    None, 1 - (gs2.top-top)],
+                         h_pad=0.45)
 
 
 @image_comparison(baseline_images=['tight_layout7'])
