Skip to content

Commit 04f9246

Browse files
committed
utilize pytest.mark.parametrize for testing
1 parent 0dc0444 commit 04f9246

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

pandas/tests/io/test_common.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -644,18 +644,16 @@ def close(self):
644644

645645

646646
@td.skip_if_no("fsspec", min_version="2023.1.0")
647-
def test_read_csv_chained_url_no_error():
647+
@pytest.mark.parametrize("compression", [None, "infer"])
648+
def test_read_csv_chained_url_no_error(compression):
648649
# GH 60100
649650
tar_file_path = "pandas/tests/io/data/tar/test-csv.tar"
650651
chained_file_url = f"tar://test.csv::file://{tar_file_path}"
651652

652-
result_a = pd.read_csv(chained_file_url, compression=None, sep=";")
653-
result_b = pd.read_csv(chained_file_url, compression="infer", sep=";")
654-
653+
result = pd.read_csv(chained_file_url, compression=compression, sep=";")
655654
expected = pd.DataFrame({"1": {0: 3}, "2": {0: 4}})
656655

657-
tm.assert_frame_equal(expected, result_a)
658-
tm.assert_frame_equal(expected, result_b)
656+
tm.assert_frame_equal(expected, result)
659657

660658

661659
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)