more test

Author: dcherian

tests/test_core.py

@@ -44,6 +44,7 @@
     assert_equal_tuple,
     has_cubed,
     has_dask,
+    has_sparse,
     raise_if_dask_computes,
     requires_cubed,
     requires_dask,
@@ -459,7 +460,18 @@ def test_numpy_reduce_nd_md():
 
 
 @requires_dask
-@pytest.mark.parametrize("reindex", [None, False, True])
+@pytest.mark.parametrize(
+    "reindex",
+    [
+        None,
+        False,
+        True,
+        pytest.param(
+            ReindexStrategy(blockwise=False, array_type=ReindexArrayType.SPARSE_COO),
+            marks=pytest.mark.skipif(not has_sparse, reason="no sparse"),
+        ),
+    ],
+)
 @pytest.mark.parametrize("func", ALL_FUNCS)
 @pytest.mark.parametrize("add_nan", [False, True])
 @pytest.mark.parametrize("dtype", (float,))
@@ -482,6 +494,9 @@ def test_groupby_agg_dask(func, shape, array_chunks, group_chunks, add_nan, dtyp
     if "arg" in func and (engine in ["flox", "numbagg"] or reindex):
         pytest.skip()
 
+    if isinstance(reindex, ReindexStrategy) and not _is_sparse_supported_reduction(func):
+        pytest.skip()
+
     rng = np.random.default_rng(12345)
     array = dask.array.from_array(rng.random(shape), chunks=array_chunks).astype(dtype)
     array = dask.array.ones(shape, chunks=array_chunks)
@@ -787,6 +802,11 @@ def test_groupby_reduce_axis_subset_against_numpy(func, axis, engine):
         (None, None),
         pytest.param(False, (2, 2, 3), marks=requires_dask),
         pytest.param(True, (2, 2, 3), marks=requires_dask),
+        pytest.param(
+            ReindexStrategy(blockwise=False, array_type=ReindexArrayType.SPARSE_COO),
+            (2, 2, 3),
+            marks=(requires_dask, pytest.mark.skipif(not has_sparse, reason="no sparse")),
+        ),
     ],
 )
 @pytest.mark.parametrize(