FIX-modin-project#2406: Attemt to fix broken test on BaseOnPython bac…

…kend This test gets a corrupted dataframe with "col2" removed by previous test cases. Signed-off-by: Gregory Shimansky <[email protected]>
gshimansky · Nov 14, 2020 · 7ce895e · 7ce895e
1 parent 966d3a4
commit 7ce895e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/modin/pandas/test/test_groupby.py b/modin/pandas/test/test_groupby.py
@@ -1194,8 +1194,14 @@ def test_shift_freq(groupby_axis, shift_axis):
     "by_and_agg_dict",
     [
         {
-            "by": ["col1", "col2"],
-            "agg_dict": {"max": ("col3", np.max), "min": ("col3", np.min)},
+            "by": [
+                list(test_data["int_data"].keys())[0],
+                list(test_data["int_data"].keys())[1],
+            ],
+            "agg_dict": {
+                "max": (list(test_data["int_data"].keys())[2], np.max),
+                "min": (list(test_data["int_data"].keys())[2], np.min),
+            },
         },
         {
             "by": ["col1"],