add basic test coverage

rapidsai · Nov 10, 2022 · 8e16f8f · 8e16f8f
1 parent 0cb6848
commit 8e16f8f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/dask_cudf/dask_cudf/tests/test_groupby.py b/python/dask_cudf/dask_cudf/tests/test_groupby.py
@@ -130,6 +130,10 @@ def test_groupby_agg(func, aggregation, pdf):
 
     assert_cudf_groupby_layers(actual)
 
+    # groupby.agg should add an explicit getitem layer
+    # to improve/enable column projection
+    assert hlg_layer(actual.dask, "getitem")
+
     dd.assert_eq(expect, actual, check_names=False, check_dtype=check_dtype)