From 17cb0256c35fcf42b0e5c6d96299adbfa19ada66 Mon Sep 17 00:00:00 2001
From: crusaderky <crusaderky@gmail.com>
Date: Fri, 22 Dec 2023 16:30:39 +0000
Subject: [PATCH] tweaks

---
 dask/tests/test_base.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/dask/tests/test_base.py b/dask/tests/test_base.py
index ee8f01b70a8..51149b11588 100644
--- 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):