From 0a9dc4f717380581a003c7b3fb84af85176bbfb4 Mon Sep 17 00:00:00 2001 From: Simon Lin Date: Wed, 24 Jul 2024 23:22:25 +1000 Subject: [PATCH] c --- py-polars/tests/unit/io/test_csv.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/py-polars/tests/unit/io/test_csv.py b/py-polars/tests/unit/io/test_csv.py index 294ed6073824..53c981cf5364 100644 --- a/py-polars/tests/unit/io/test_csv.py +++ b/py-polars/tests/unit/io/test_csv.py @@ -580,11 +580,8 @@ def test_compressed_csv(io_files_path: Path, monkeypatch: pytest.MonkeyPatch) -> # zstd compressed file csv_file = io_files_path / "zstd_compressed.csv.zst" - with pytest.raises( - ComputeError, - match="cannot scan compressed csv; use `read_csv` for compressed data", - ): - pl.scan_csv(csv_file).collect() + out = pl.scan_csv(csv_file, truncate_ragged_lines=True).collect() + assert_frame_equal(out, expected) out = pl.read_csv(str(csv_file), truncate_ragged_lines=True) assert_frame_equal(out, expected)