reduce test size

Eventual-Inc · Oct 25, 2024 · 3485573 · 3485573
1 parent cffe79a
commit 3485573
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/io/test_write_modes.py b/tests/io/test_write_modes.py
@@ -64,13 +64,13 @@ def arrange_write_mode_test(existing_data, new_data, path, format, write_mode, p
 
 @pytest.mark.parametrize("write_mode", ["append", "overwrite"])
 @pytest.mark.parametrize("format", ["csv", "parquet"])
-@pytest.mark.parametrize("num_partitions", [1, 50, 100])
+@pytest.mark.parametrize("num_partitions", [1, 2])
 @pytest.mark.parametrize("partition_cols", [None, ["a"]])
 def test_write_modes_local(tmp_path, write_mode, format, num_partitions, partition_cols):
     path = str(tmp_path)
-    existing_data = {"a": [i for i in range(100)]}
+    existing_data = {"a": [i for i in range(10)]}
     new_data = {
-        "a": [i for i in range(100, 200)],
+        "a": [i for i in range(10, 20)],
     }
 
     read_back = arrange_write_mode_test(