add test

rapidsai · Sep 18, 2023 · 9becd63 · 9becd63
1 parent e95830c
commit 9becd63
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/python/cudf/cudf/tests/test_multiindex.py b/python/cudf/cudf/tests/test_multiindex.py
@@ -2009,3 +2009,12 @@ def test_multiindex_to_frame_allow_duplicates(
                 )
 
             assert_eq(expected, actual)
+
+
+def test_multiIndex_codes():
+    midx = cudf.MultiIndex.from_tuples(
+        [("a", "b"), ("a", "c"), ("b", "c")], names=["A", "Z"]
+    )
+
+    for p_array, g_array in zip(midx.to_pandas().codes, midx.codes):
+        assert_eq(p_array, g_array)