refactor(): Small corrections on unit tests

UFRN-URNAI · Oct 8, 2024 · 724176e · 724176e
1 parent cc53ee1
commit 724176e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/units/models/test_model_base.py b/tests/units/models/test_model_base.py
@@ -17,9 +17,9 @@ def __init__(self):
 		learn_return = fake_model.learn("current_state",
 			"action", "reward", "next_state")
 		predict_return = fake_model.predict("state")
-		learning_dict = fake_model.learning_dict
+		learning_data = fake_model.learning_data
 
 		assert isinstance(ModelBase, ABCMeta)
 		assert learn_return is None
 		assert predict_return is None
-		assert learning_dict is None
+		self.assertEqual(learning_data, {})
diff --git a/tests/units/sc2/actions/test_sc2_action.py b/tests/units/sc2/actions/test_sc2_action.py
@@ -2,7 +2,7 @@
 
 from pysc2.lib import actions
 
-from urnai.actions.sc2_action import SC2Action
+from urnai.sc2.actions.sc2_action import SC2Action
 
 _BUILD_REFINERY = actions.RAW_FUNCTIONS.Build_Refinery_pt
 _NO_OP = actions.FUNCTIONS.no_op