From 3485573f5fbaa20604bb75915fac1b3a49d467ba Mon Sep 17 00:00:00 2001 From: Colin Ho Date: Fri, 25 Oct 2024 08:55:56 -0700 Subject: [PATCH] reduce test size --- tests/io/test_write_modes.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/io/test_write_modes.py b/tests/io/test_write_modes.py index 820e25c448..173a7899a5 100644 --- 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(