Update tests for GroupBy.value_counts

rapidsai · Sep 18, 2023 · 7c91283 · 7c91283
1 parent 8164c5a
commit 7c91283
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/cudf/cudf/tests/test_groupby.py b/python/cudf/cudf/tests/test_groupby.py
@@ -3499,7 +3499,9 @@ def test_group_by_value_counts(normalize, sort, ascending, dropna, as_index):
     )
 
     # TODO: Remove `check_names=False` once testing against `pandas>=2.0.0`
-    assert_eq(actual, expected, check_names=False)
+    assert_groupby_results_equal(
+        actual, expected, check_names=False, check_index_type=False
+    )
 
 
 def test_group_by_value_counts_subset():
@@ -3517,4 +3519,6 @@ def test_group_by_value_counts_subset():
     expected = pdf.groupby("gender").value_counts(["education"])
 
     # TODO: Remove `check_names=False` once testing against `pandas>=2.0.0`
-    assert_eq(actual, expected, check_names=False)
+    assert_groupby_results_equal(
+        actual, expected, check_names=False, check_index_type=False
+    )