tweaks

mrocklin · Dec 22, 2023 · 17cb025 · 17cb025
1 parent 3d35e88
commit 17cb025
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dask/tests/test_base.py b/dask/tests/test_base.py
@@ -710,11 +710,14 @@ def optimize(self, fuse=False):
             return self
 
     dyf = dx._collection.new_collection(DoNotMaterialize())
-    assert is_dask_collection(dyf)
 
+    with pytest.raises(AssertionError, match="must not reach"):
+        dyf.__dask_keys__()
     with pytest.raises(AssertionError, match="must not reach"):
         dyf.__dask_graph__()
 
+    assert is_dask_collection(dyf)
+
 
 def test_unpack_collections():
     class ANamedTuple(NamedTuple):