test: deduplicate pytest params for problem 1729

tablexpr · Jan 21, 2025 · a1c2661 · a1c2661
1 parent f67e10b
commit a1c2661
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 34 deletions.
diff --git a/tests/test_pandas.py b/tests/test_pandas.py
@@ -1082,23 +1082,7 @@ def test_problem_1683(input_data, expected_data):
 
 @pytest.mark.parametrize(
     "input_data, expected_data",
-    [
-        pytest.param(
-            {"user_id": [1, 1, 2, 2, 3], "follower_id": [10, 11, 12, 13, 14]},
-            {"user_id": [1, 2, 3], "followers_count": [2, 2, 1]},
-            id="multiple_users",
-        ),
-        pytest.param(
-            {"user_id": [1], "follower_id": [10]},
-            {"user_id": [1], "followers_count": [1]},
-            id="single_user",
-        ),
-        pytest.param(
-            {"user_id": [1, 1, 1], "follower_id": [10, 11, 12]},
-            {"user_id": [1], "followers_count": [3]},
-            id="single_user_multiple_followers",
-        ),
-    ],
+    PARAMS_PROBLEM_1729,
 )
 def test_problem_1729(input_data, expected_data):
     table = pd.DataFrame(input_data)

diff --git a/tests/test_problem_params.py b/tests/test_problem_params.py
@@ -1227,6 +1227,24 @@
     ),
 ]
 
+PARAMS_PROBLEM_1729 = [
+    pytest.param(
+        {"user_id": [1, 1, 2, 2, 3], "follower_id": [10, 11, 12, 13, 14]},
+        {"user_id": [1, 2, 3], "followers_count": [2, 2, 1]},
+        id="multiple_users",
+    ),
+    pytest.param(
+        {"user_id": [1], "follower_id": [10]},
+        {"user_id": [1], "followers_count": [1]},
+        id="single_user",
+    ),
+    pytest.param(
+        {"user_id": [1, 1, 1], "follower_id": [10, 11, 12]},
+        {"user_id": [1], "followers_count": [3]},
+        id="single_user_multiple_followers",
+    ),
+]
+
 PARAMS_PROBLEM_1757 = [
     (
         {

diff --git a/tests/test_pyarrow.py b/tests/test_pyarrow.py
@@ -972,23 +972,7 @@ def test_problem_1683(input_data, expected_data):
 
 @pytest.mark.parametrize(
     "input_data, expected_data",
-    [
-        pytest.param(
-            {"user_id": [1, 1, 2, 2, 3], "follower_id": [10, 11, 12, 13, 14]},
-            {"user_id": [1, 2, 3], "followers_count": [2, 2, 1]},
-            id="multiple_users",
-        ),
-        pytest.param(
-            {"user_id": [1], "follower_id": [10]},
-            {"user_id": [1], "followers_count": [1]},
-            id="single_user",
-        ),
-        pytest.param(
-            {"user_id": [1, 1, 1], "follower_id": [10, 11, 12]},
-            {"user_id": [1], "followers_count": [3]},
-            id="single_user_multiple_followers",
-        ),
-    ],
+    PARAMS_PROBLEM_1729,
 )
 def test_problem_1729(input_data, expected_data):
     table = pa.Table.from_pydict(input_data)