[dev] make sure graphbolt is not imported together with dgl (#7756)

dmlc · Aug 29, 2024 · 03e83ac · 03e83ac
1 parent 9782c02
commit 03e83ac
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/tests/python/test_dgl_import.py b/tests/python/test_dgl_import.py
@@ -0,0 +1,10 @@
+import sys
+
+
+def test_graphbolt_is_not_imported():
+    assert (
+        "dgl.graphbolt" not in sys.modules
+    ), "dgl.graphbolt is already imported"
+    import dgl
+
+    assert "dgl.graphbolt" not in sys.modules, "dgl.graphbolt is imported"
diff --git a/tests/scripts/task_unit_test.sh b/tests/scripts/task_unit_test.sh
@@ -39,6 +39,7 @@ if [ $DGLBACKEND == "mxnet" ]
 then
   python3 -m pytest -v --junitxml=pytest_compute.xml --durations=100 --ignore=tests/python/common/test_ffi.py tests/python/common || fail "common"
 else
+  python3 -m pytest -v --junitxml=pytest_dgl_import.xml tests/python/test_dgl_import.py || fail "dgl_import"
   python3 -m pytest -v --junitxml=pytest_common.xml --durations=100 tests/python/common || fail "common"
 fi
 python3 -m pytest -v --junitxml=pytest_backend.xml --durations=100 tests/python/$DGLBACKEND || fail "backend-specific"