fixing spark session weirdness with parameters

capitalone · May 10, 2024 · cda8ccc · cda8ccc
1 parent 8834a3b
commit cda8ccc
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/tests/test_spark.py b/tests/test_spark.py
@@ -1207,9 +1207,8 @@ def test_dupes_with_nulls_ints():
 
 
 @pandas_version
-@pytest.mark.parametrize(
-    "dataframe,expected",
-    [
+def test_generate_id_within_group():
+    matrix = [
         (ps.DataFrame({"a": [1, 2, 3], "b": [1, 2, 3]}), ps.Series([0, 0, 0])),
         (
             ps.DataFrame({"a": ["a", "a", "DATACOMPY_NULL"], "b": [1, 1, 2]}),
@@ -1230,10 +1229,11 @@ def test_dupes_with_nulls_ints():
             ),
             ps.Series([0, 0, 1]),
         ),
-    ],
-)
-def test_generate_id_within_group(dataframe, expected):
-    assert (generate_id_within_group(dataframe, ["a", "b"]) == expected).all()
+]
+    for i in matrix:
+        dataframe = i[0]
+        expected = i[1]
+        assert (generate_id_within_group(dataframe, ["a", "b"]) == expected).all()
 
 
 @pandas_version