oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#1714: fix tests

Browse Source
This commit is contained in:
Czar Echavez
2023-08-03 12:42:16 +01:00
parent 483fa7d841
commit b08683fcd3
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@@ -37,7 +37,7 @@ dependencies = [
"stable-baselines3==1.6.2", "stable-baselines3==1.6.2",
"tensorflow==2.12.0", "tensorflow==2.12.0",
"typer[all]==0.9.0", "typer[all]==0.9.0",
"pydantic" "pydantic==2.1.1"
] ]
[tool.setuptools.dynamic] [tool.setuptools.dynamic]

2
tests/conftest.py
View File

@@ -96,7 +96,7 @@ def temp_primaite_session(request):
""" """
training_config_path = request.param[0] training_config_path = request.param[0]
lay_down_config_path = request.param[1] lay_down_config_path = request.param[1]
with patch("_primaite.agents.agent_abc.get_session_path", get_temp_session_path) as mck: with patch("primaite.agents.agent_abc.get_session_path", get_temp_session_path) as mck:
mck.session_timestamp = datetime.now() mck.session_timestamp = datetime.now()
return TempPrimaiteSession(training_config_path, lay_down_config_path) return TempPrimaiteSession(training_config_path, lay_down_config_path)

2
tests/unit_tests/_primaite/_simulator/test_core.py
View File

@@ -43,7 +43,7 @@ class TestIsolatedSimComponent:
comp = TestComponent(name="computer", size=(5, 10)) comp = TestComponent(name="computer", size=(5, 10))
dump = comp.model_dump() dump = comp.model_dump()
assert dump == {"name": "computer", "size": (5, 10)} assert dump["name"] is "computer"
def test_apply_action(self): def test_apply_action(self):
"""Validate that we can override apply_action behaviour and it updates the state of the component.""" """Validate that we can override apply_action behaviour and it updates the state of the component."""