Skip to content

Commit 32fef29

Browse files
committed
use tm.assert_frame_equal; add separator on read_csv; improve chained_file_url
1 parent deb21df commit 32fef29

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

pandas/tests/io/test_common.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -646,16 +646,16 @@ def close(self):
646646
@td.skip_if_no("fsspec")
647647
def test_read_csv_chained_url_no_error():
648648
# GH 60100
649-
tar_file = "pandas/tests/io/data/tar/test-csv.tar"
649+
tar_file_path = "pandas/tests/io/data/tar/test-csv.tar"
650+
chained_file_url = f"tar://test.csv::file://{tar_file_path}"
650651

651-
x = pd.read_csv(f"tar://test.csv::file://{tar_file}", compression=None)
652-
y = pd.read_csv(f"tar://test.csv::file://{tar_file}", compression="infer")
652+
result_a = pd.read_csv(chained_file_url, compression=None, sep=";")
653+
result_b = pd.read_csv(chained_file_url, compression="infer", sep=";")
653654

654-
x_to_json_expected_output = '{"1;2":{"0":"3;4"}}'
655-
y_to_json_expected_output = '{"1;2":{"0":"3;4"}}'
655+
expected = pd.DataFrame({"1": {0: 3}, "2": {0: 4}})
656656

657-
assert x_to_json_expected_output == x.to_json()
658-
assert y_to_json_expected_output == y.to_json()
657+
tm.assert_frame_equal(expected, result_a)
658+
tm.assert_frame_equal(expected, result_b)
659659

660660

661661
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)