test: deduplicate pytest params for problem 1141

Author: IndexSeek

tests/test_pandas.py

@@ -310,38 +310,7 @@ def test_problem_1075(input_data_1, input_data_2, expected_data):
 
 @pytest.mark.parametrize(
     "input_data, expected_data",
-    [
-        pytest.param(
-            {
-                "user_id": [1, 1, 1, 2, 2, 2],
-                "session_id": [1, 1, 1, 4, 4, 4],
-                "activity_date": [
-                    datetime(2019, 7, 20, 0, 0),
-                    datetime(2019, 7, 20, 0, 0),
-                    datetime(2019, 7, 20, 0, 0),
-                    datetime(2019, 7, 21, 0, 0),
-                    datetime(2019, 7, 21, 0, 0),
-                    datetime(2019, 7, 21, 0, 0),
-                ],
-                "activity_type": [
-                    "open_session",
-                    "scroll_down",
-                    "end_session",
-                    "open_session",
-                    "send_message",
-                    "end_session",
-                ],
-            },
-            {
-                "day": [
-                    datetime(2019, 7, 20, 0, 0),
-                    datetime(2019, 7, 21, 0, 0),
-                ],
-                "active_users": [1, 1],
-            },
-            id="happy_path_1",
-        )
-    ],
+    PARAMS_PROBLEMS_1141,
 )
 def test_problem_1141(input_data, expected_data):
     table = pd.DataFrame(input_data)

tests/test_problem_params.py

@@ -839,6 +839,39 @@
 ]
 
 
+PARAMS_PROBLEMS_1141 = [
+    pytest.param(
+        {
+            "user_id": [1, 1, 1, 2, 2, 2],
+            "session_id": [1, 1, 1, 4, 4, 4],
+            "activity_date": [
+                datetime(2019, 7, 20, 0, 0),
+                datetime(2019, 7, 20, 0, 0),
+                datetime(2019, 7, 20, 0, 0),
+                datetime(2019, 7, 21, 0, 0),
+                datetime(2019, 7, 21, 0, 0),
+                datetime(2019, 7, 21, 0, 0),
+            ],
+            "activity_type": [
+                "open_session",
+                "scroll_down",
+                "end_session",
+                "open_session",
+                "send_message",
+                "end_session",
+            ],
+        },
+        {
+            "day": [
+                datetime(2019, 7, 20, 0, 0),
+                datetime(2019, 7, 21, 0, 0),
+            ],
+            "active_users": [1, 1],
+        },
+        id="happy_path_1",
+    )
+]
+
 PARAMS_PROBLEM_1164 = [
     pytest.param(
         {

tests/test_pyarrow.py

@@ -296,38 +296,7 @@ def test_problem_1075(input_data_1, input_data_2, expected_data):
 
 @pytest.mark.parametrize(
     "input_data, expected_data",
-    [
-        pytest.param(
-            {
-                "user_id": [1, 1, 1, 2, 2, 2],
-                "session_id": [1, 1, 1, 4, 4, 4],
-                "activity_date": [
-                    datetime(2019, 7, 20, 0, 0),
-                    datetime(2019, 7, 20, 0, 0),
-                    datetime(2019, 7, 20, 0, 0),
-                    datetime(2019, 7, 21, 0, 0),
-                    datetime(2019, 7, 21, 0, 0),
-                    datetime(2019, 7, 21, 0, 0),
-                ],
-                "activity_type": [
-                    "open_session",
-                    "scroll_down",
-                    "end_session",
-                    "open_session",
-                    "send_message",
-                    "end_session",
-                ],
-            },
-            {
-                "day": [
-                    datetime(2019, 7, 20, 0, 0),
-                    datetime(2019, 7, 21, 0, 0),
-                ],
-                "active_users": [1, 1],
-            },
-            id="happy_path_1",
-        )
-    ],
+    PARAMS_PROBLEMS_1141,
 )
 def test_problem_1141(input_data, expected_data):
     input_table = pa.Table.from_pydict(input_data)