TST: remove yield warnings from test_windows.py

Author: jreback

pandas/tests/formats/test_format.py

@@ -1392,24 +1392,26 @@ def test_repr_html_long(self):
         assert u('2 columns') in long_repr
 
     def test_repr_html_float(self):
-        max_rows = get_option('display.max_rows')
-        h = max_rows - 1
-        df = DataFrame({'idx': np.linspace(-10, 10, h),
-                        'A': np.arange(1, 1 + h),
-                        'B': np.arange(41, 41 + h)}).set_index('idx')
-        reg_repr = df._repr_html_()
-        assert '..' not in reg_repr
-        assert str(40 + h) in reg_repr
-
-        h = max_rows + 1
-        df = DataFrame({'idx': np.linspace(-10, 10, h),
-                        'A': np.arange(1, 1 + h),
-                        'B': np.arange(41, 41 + h)}).set_index('idx')
-        long_repr = df._repr_html_()
-        assert '..' in long_repr
-        assert '31' not in long_repr
-        assert u('%d rows ') % h in long_repr
-        assert u('2 columns') in long_repr
+        with option_context('display.max_rows', 60):
+
+            max_rows = get_option('display.max_rows')
+            h = max_rows - 1
+            df = DataFrame({'idx': np.linspace(-10, 10, h),
+                            'A': np.arange(1, 1 + h),
+                            'B': np.arange(41, 41 + h)}).set_index('idx')
+            reg_repr = df._repr_html_()
+            assert '..' not in reg_repr
+            assert str(40 + h) in reg_repr
+
+            h = max_rows + 1
+            df = DataFrame({'idx': np.linspace(-10, 10, h),
+                            'A': np.arange(1, 1 + h),
+                            'B': np.arange(41, 41 + h)}).set_index('idx')
+            long_repr = df._repr_html_()
+            assert '..' in long_repr
+            assert '31' not in long_repr
+            assert u('%d rows ') % h in long_repr
+            assert u('2 columns') in long_repr
 
     def test_repr_html_long_multiindex(self):
         max_rows = get_option('display.max_rows')

pandas/tests/test_window.py

@@ -646,7 +646,7 @@ def test_dtypes(self):
             f = self.funcs[f_name]
             d = self.data[d_name]
             exp = self.expects[d_name][f_name]
-            yield self.check_dtypes, f, f_name, d, d_name, exp
+            self.check_dtypes(f, f_name, d, d_name, exp)
 
     def check_dtypes(self, f, f_name, d, d_name, exp):
         roll = d.rolling(window=self.window)