Skip to content

Commit 150e6e6

Browse files
committed
Use less data in test_to_csv_chunking
1 parent 169830f commit 150e6e6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pandas/tests/frame/methods/test_to_csv.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -836,9 +836,9 @@ def test_to_csv_dups_cols2(self, temp_file):
836836
result = result.rename(columns={"a.1": "a"})
837837
tm.assert_frame_equal(result, df)
838838

839-
@pytest.mark.parametrize("chunksize", [10000, 50000, 100000])
839+
@pytest.mark.parametrize("chunksize", [1, 5, 10])
840840
def test_to_csv_chunking(self, chunksize, temp_file):
841-
aa = DataFrame({"A": range(100000)})
841+
aa = DataFrame({"A": range(10)})
842842
aa["B"] = aa.A + 1.0
843843
aa["C"] = aa.A + 2.0
844844
aa["D"] = aa.A + 3.0

0 commit comments

Comments
 (0)