Fix freq behavior on ci (#1092)

dask · Jun 25, 2024 · d0e0c21 · d0e0c21
1 parent cb121cd
commit d0e0c21
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/dask_expr/tests/test_groupby.py b/dask_expr/tests/test_groupby.py
@@ -795,15 +795,21 @@ def func(df):
 def test_groupby_index_array(pdf):
     pdf.index = pd.date_range(start="2020-12-31", freq="D", periods=len(pdf))
     df = from_pandas(pdf, npartitions=10)
+    # pandas loses the freq for mac but keeps it on ubuntu
+    expected = pdf.x.groupby(pdf.index).nunique()
+    expected.index.freq = None
 
     assert_eq(
         df.x.groupby(df.index).nunique(),
-        pdf.x.groupby(pdf.index).nunique(),
+        expected,
         check_names=False,
     )
+    # pandas loses the freq for mac but keeps it on ubuntu
+    expected = pdf.groupby(pdf.index).x.nunique()
+    expected.index.freq = None
     assert_eq(
         df.groupby(df.index).x.nunique(),
-        pdf.groupby(pdf.index).x.nunique(),
+        expected,
         check_names=False,
     )