harden test again shifts in basic column touching

dask-contrib · Feb 12, 2024 · ba147c3 · ba147c3
1 parent 7df3cdb
commit ba147c3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_inspect.py b/tests/test_inspect.py
@@ -97,7 +97,10 @@ def test_basic_root_works():
         .fill(events.MET_pt)
     )
 
+    columns = list(dak.necessary_columns(q1_hist).values())[0]
+    assert columns == frozenset({"MET_pt"})
     dask.compute(q1_hist)
+
 
 
 def test_sample(daa):