ingore order

rapidsai · Apr 25, 2024 · 68742a6 · 68742a6
1 parent 88e7da8
commit 68742a6
Showing 1 changed file with 2 additions and 13 deletions.
diff --git a/python/cuxfilter/tests/charts/datashader/test_graph_assets.py b/python/cuxfilter/tests/charts/datashader/test_graph_assets.py
@@ -1,8 +1,8 @@
-import dask_cudf
 import pytest
 import cudf
 import cupy as cp
 from cuxfilter.charts.datashader.custom_extensions import graph_assets
+from dask.dataframe import assert_eq
 
 from ..utils import initialize_df, df_types
 
@@ -50,15 +50,4 @@ def test_calc_connected_edges(
             node_y_dtype=cp.float32,
         ).reset_index(drop=True)
 
-        res = (
-            res.compute().reset_index(drop=True)
-            if isinstance(res, dask_cudf.DataFrame)
-            else res
-        )
-        result = (
-            result.compute()
-            if isinstance(result, dask_cudf.DataFrame)
-            else result
-        )
-
-        assert res.to_pandas().equals(result.to_pandas())
+        assert_eq(res, result, check_divisions=False, check_index=False)