fix: Move an import which is dependent on the OS but only runs when t…

…esting into the test util function (#675)
FlexMeasures · May 9, 2023 · 47248d1 · 47248d1
1 parent be8a88e
commit 47248d1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/.vscode/spellright.dict b/.vscode/spellright.dict
@@ -250,3 +250,4 @@ Changelog
 Bugfixes
 Dockerfile
 ―
+nt
diff --git a/flexmeasures/data/tests/utils.py b/flexmeasures/data/tests/utils.py
@@ -2,13 +2,15 @@
 
 import click
 
-if os.name == "nt":
-    from rq_win import WindowsWorker as SimpleWorker
-else:
-    from rq import SimpleWorker
-
 
 def work_on_rq(redis_queue, exc_handler=None):
+
+    #  we only want this import distinction to matter when we actually are testing
+    if os.name == "nt":
+        from rq_win import WindowsWorker as SimpleWorker
+    else:
+        from rq import SimpleWorker
+
     exc_handlers = []
     if exc_handler is not None:
         exc_handlers.append(exc_handler)