Skip to content

Commit 830d040

Browse files
Small corrections on unit tests (#104)
refactor(): Small corrections on unit tests
1 parent cc53ee1 commit 830d040

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/units/models/test_model_base.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@ def __init__(self):
1717
learn_return = fake_model.learn("current_state",
1818
"action", "reward", "next_state")
1919
predict_return = fake_model.predict("state")
20-
learning_dict = fake_model.learning_dict
20+
learning_data = fake_model.learning_data
2121

2222
assert isinstance(ModelBase, ABCMeta)
2323
assert learn_return is None
2424
assert predict_return is None
25-
assert learning_dict is None
25+
self.assertEqual(learning_data, {})

tests/units/sc2/actions/test_sc2_action.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from pysc2.lib import actions
44

5-
from urnai.actions.sc2_action import SC2Action
5+
from urnai.sc2.actions.sc2_action import SC2Action
66

77
_BUILD_REFINERY = actions.RAW_FUNCTIONS.Build_Refinery_pt
88
_NO_OP = actions.FUNCTIONS.no_op

0 commit comments

Comments
 (0)