Fix for loading nodes when their module is not in the pythonpath at the time of loading it

Author: PaulSchweizer

flowpipe/utilities.py

@@ -19,6 +19,8 @@ def import_class(module, cls_name, file_location=None):
         module = importlib.import_module(module)
     except NameError:  # pragma: no cover
         module = __import__(module, globals(), locals(), ["object"], -1)
+    except ModuleNotFoundError:  # pragma: no cover
+        pass  # this exception will be ignored to force the Source File load further down
     try:
         cls = getattr(module, cls_name)
     except AttributeError:  # pragma: no cover

tests/test_utilities.py

@@ -5,9 +5,11 @@
 import sys
 from hashlib import sha256
 
+import mock
 import numpy as np
 
 import flowpipe.utilities as util
+from flowpipe import INode
 
 
 class WeirdObject(object):
@@ -93,3 +95,15 @@ def test_sanitize_string():
     expected_string = "This is a {{test}} string"
     sanitized_string = util.sanitize_string_input(test_string)
     assert sanitized_string == expected_string
+
+
+class NodeInTest(INode):
+    """"""
+
+
+@mock.patch("importlib.import_module", side_effect=ModuleNotFoundError)
+def test_import_class_from_current_file(_):
+    cls = util.import_class(
+        "test_utilities", "NodeInTest", file_location=__file__
+    )
+    assert cls.__name__ == NodeInTest.__name__