add dragon utils to pythonpath

CrayLabs · Oct 30, 2024 · b8b0d41 · b8b0d41
1 parent 7ea00ab
commit b8b0d41
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -58,8 +58,9 @@ force-exclude = '''
 [tool.pytest.ini_options]
 log_cli = true
 log_cli_level = "debug"
+pythonpath = ["tests/dragon_wlm_/utils"]
 testpaths = [
-    "tests"
+    "tests",
 ]
 norecursedirs = [
     "tests/test_configs",

diff --git a/tests/dragon_wlm/test_request_dispatcher.py b/tests/dragon_wlm/test_request_dispatcher.py
@@ -75,8 +75,8 @@
 from smartsim._core.mli.message_handler import MessageHandler
 from smartsim.log import get_logger
 
-from .channel import FileSystemCommChannel
-from .msg_pump import mock_messages
+from .utils.channel import FileSystemCommChannel
+from .utils.msg_pump import mock_messages
 
 logger = get_logger(__name__)
 

diff --git a/tests/dragon_wlm/msg_pump.py → tests/dragon_wlm/utils/msg_pump.py b/tests/dragon_wlm/msg_pump.py → tests/dragon_wlm/utils/msg_pump.py