parametrize write benchmark

python/deltalake/writer.py

@@ -125,7 +125,7 @@ def write_deltalake(
         table_or_uri: URI of a table or a DeltaTable object.
         data: Data to write. If passing iterable, the schema must also be given.
         schema: Optional schema to write.
-        partition_by: List of columns to partition the table by. Only required 
+        partition_by: List of columns to partition the table by. Only required
             when creating a new table.
         filesystem: Optional filesystem to pass to PyArrow. If not provided will
             be inferred from uri. The file system has to be rooted in the table root.

python/tests/test_benchmark.py

@@ -24,9 +24,12 @@ def sample_table() -> pa.Table:
     return tab
 
 
+@pytest.mark.parametrize("engine", ["pyarrow", "rust"])
 @pytest.mark.benchmark(group="write")
-def test_benchmark_write(benchmark, sample_table, tmp_path):
-    benchmark(write_deltalake, str(tmp_path), sample_table, mode="overwrite")
+def test_benchmark_write(benchmark, sample_table, tmp_path, engine):
+    benchmark(
+        write_deltalake, str(tmp_path), sample_table, mode="overwrite", engine=engine
+    )
 
     dt = DeltaTable(str(tmp_path))
     assert dt.to_pyarrow_table().sort_by("i") == sample_table