Ensure tests run even if dask is not installed

jupyter · Dec 29, 2023 · 739cb75 · 739cb75
1 parent 3de0911
commit 739cb75
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/papyri/tests/test_gen.py b/papyri/tests/test_gen.py
@@ -206,12 +206,21 @@ def test_self_2():
     g.collect_api_docs(
         "papyri", {"papyri", "papyri.take2:RefInfo", "papyri.take2:RefInfo.__eq__"}
     )
-    assert (
-        g.data["papyri"].to_dict()["arbitrary"][4]["children"][1]["children"][0]["dt"][
-            "children"
-        ][0]["reference"]["module"]
-        == "dask"
-    )
+    try:
+        import dask
+        assert (
+            g.data["papyri"].to_dict()["arbitrary"][4]["children"][1]["children"][0]["dt"][
+                "children"
+            ][0]["reference"]["module"]
+            == "dask"
+        )
+    except ImportError:
+        assert (
+            g.data["papyri"].to_dict()["arbitrary"][4]["children"][1]["children"][0]["dt"][
+                "children"
+            ][0]["domain"]
+            is None
+        )
 
     assert (
         g.data["papyri.take2:RefInfo"]