nrows is now parametrized

Author: roberthdevries

pandas/tests/io/parser/test_textreader.py

@@ -341,16 +341,16 @@ def test_empty_csv_input(self):
         df = read_csv(StringIO(), chunksize=20, header=None, names=["a", "b", "c"])
         assert isinstance(df, TextFileReader)
 
-    def test_blank_lines_between_header_and_data_rows(self):
+    @pytest.mark.parametrize("nrows", range(1,6))
+    def test_blank_lines_between_header_and_data_rows(self, nrows):
         # GH 28071
         ref = DataFrame(
             [[np.nan, np.nan], [np.nan, np.nan], [1, 2], [np.nan, np.nan], [3, 4]],
             columns=list("ab"),
         )
         csv = "\nheader\n\na,b\n\n\n1,2\n\n3,4"
-        for nrows in range(1, 6):
-            df = read_csv(StringIO(csv), header=3, nrows=nrows, skip_blank_lines=False)
-            tm.assert_frame_equal(df, ref[:nrows])
+        df = read_csv(StringIO(csv), header=3, nrows=nrows, skip_blank_lines=False)
+        tm.assert_frame_equal(df, ref[:nrows])
 
 
 def assert_array_dicts_equal(left, right):