test: update parquet compression tests to validate invalid and default compression levels

Author: kosiew

python/tests/test_dataframe.py

@@ -1105,20 +1105,20 @@ def test_write_compressed_parquet_wrong_compression_level(
         )
 
 
-# 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):
+@pytest.mark.parametrize("compression", ["wrong"])
+def test_write_compressed_parquet_invalid_compression(df, tmp_path, compression):
     path = tmp_path
 
-    df.write_parquet(str(path), compression=compression)
+    with pytest.raises(ValueError):
+        df.write_parquet(str(path), compression=compression)
 
 
-@pytest.mark.parametrize("compression", ["wrong"])
-def test_write_compressed_parquet_missing_compression_level(df, tmp_path, compression):
+# 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
 
-    with pytest.raises(ValueError):
-        df.write_parquet(str(path), compression=compression)
+    df.write_parquet(str(path), compression=compression)
 
 
 def test_dataframe_export(df) -> None: