Update python/cudf/cudf/tests/test_groupby.py

rapidsai · Oct 15, 2021 · 0dde507 · 0dde507
1 parent ee30e27
commit 0dde507
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf/tests/test_groupby.py b/python/cudf/cudf/tests/test_groupby.py
@@ -1961,7 +1961,7 @@ def test_groupby_diff_row_mixed_numerics(nelem, shift_perc, direction):
     got = gdf.groupby(["0"]).diff(periods=n_shift)
 
     assert_groupby_results_equal(
-        expected[["1", "2", "3", "4"]], got[["1", "2", "3", "4"]]
+        expected[["1", "2", "3", "4", "5"]], got[["1", "2", "3", "4", "5"]]
     )