python parquet: default to lz4 compression (#3129)

* python parquet: default to lz4 compression * temp fix test
pola-rs · Apr 12, 2022 · ff29698 · ff29698
1 parent 3708721
commit ff29698
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/py-polars/polars/internals/frame.py b/py-polars/polars/internals/frame.py
@@ -1335,7 +1335,7 @@ def write_parquet(
                 ],
                 str,
             ]
-        ] = "snappy",
+        ] = "lz4",
         statistics: bool = False,
         use_pyarrow: bool = False,
         **kwargs: Any,

diff --git a/py-polars/tests/io/test_parquet.py b/py-polars/tests/io/test_parquet.py
@@ -126,7 +126,8 @@ def test_parquet_datetime() -> None:
     df = pl.DataFrame(data)
     df = df.with_column(df["datetime"].cast(pl.Datetime))
 
-    df.write_parquet(f, use_pyarrow=True)
+    # todo! test all compressions here
+    df.write_parquet(f, use_pyarrow=True, compression="snappy")
     f.seek(0)
     read = pl.read_parquet(f)
     assert read.frame_equal(df)