Fix groupby_reduce

xarray-contrib · Oct 12, 2022 · bed84c2 · bed84c2
1 parent a9c7cae
commit bed84c2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -140,7 +140,9 @@ def test_groupby_reduce(
         split_out=split_out,
         engine=engine,
     )
-    g_dtype = by.dtype if expected_groups is None else np.asarray(expected_groups).dtype
+    # we use pd.Index(expected_groups).to_numpy() which is always int64
+    # for the values in this test
+    g_dtype = by.dtype if expected_groups is None else np.int64
 
     assert_equal(groups, np.array([0, 1, 2], g_dtype))
     assert_equal(expected_result, result)