add test

Author: kosiew

python/tests/test_dataframe.py

@@ -731,9 +731,7 @@ def test_optimized_logical_plan(aggregate_df):
 def test_execution_plan(aggregate_df):
     plan = aggregate_df.execution_plan()
 
-    expected = (
-        "AggregateExec: mode=FinalPartitioned, gby=[c1@0 as c1], aggr=[sum(test.c2)]\n"  # noqa: E501
-    )
+    expected = "AggregateExec: mode=FinalPartitioned, gby=[c1@0 as c1], aggr=[sum(test.c2)]\n"  # noqa: E501
 
     assert expected == plan.display()
 
@@ -1107,7 +1105,15 @@ def test_write_compressed_parquet_wrong_compression_level(
         )
 
 
-@pytest.mark.parametrize("compression", ["brotli", "zstd", "wrong"])
+# test write_parquet with zstd, brotli default compression level, should complete without error
+@pytest.mark.parametrize("compression", ["zstd", "brotli"])
+def test_write_compressed_parquet_default_compression_level(df, tmp_path, compression):
+    path = tmp_path
+
+    df.write_parquet(str(path), compression=compression)
+
+
+@pytest.mark.parametrize("compression", ["wrong"])
 def test_write_compressed_parquet_missing_compression_level(df, tmp_path, compression):
     path = tmp_path