fix test in python 3.6

RasaHQ · Jan 14, 2021 · 7432444 · 7432444
1 parent 3628341
commit 7432444
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/shared/utils/test_common.py b/tests/shared/utils/test_common.py
@@ -5,8 +5,8 @@
 import pytest
 from _pytest.recwarn import WarningsRecorder
 
+import rasa.shared.core.domain
 import rasa.shared.utils.common
-from rasa.shared.core.domain import Domain, State
 
 
 def test_all_subclasses():
@@ -178,8 +178,8 @@ def test_class_from_module_path_not_found(
 @pytest.mark.parametrize(
     "module_path, result, outcome",
     [
-        ("rasa.shared.core.domain.Domain", Domain, True),
-        ("rasa.shared.core.domain.State", State, False),
+        ("rasa.shared.core.domain.Domain", rasa.shared.core.domain.Domain, True),
+        ("rasa.shared.core.domain.logger", rasa.shared.core.domain.logger, False),
     ],
 )
 def test_class_from_module_path_ensure_class(