steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-07-06 01:21:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

maintainer test grabs temp state

Browse Source
This commit is contained in:
Chanumask
2023-11-23 12:52:27 +01:00
parent 3a7b727ec6
commit 4c85cac50f
9 changed files with 82 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
marl_factory_grid/environment/tests.py
View File

@ -1,10 +1,10 @@
from typing import List
import unittest
from typing import List
from marl_factory_grid.modules import Door, Machine, Maintainer
from marl_factory_grid.utils.results import TickResult, DoneResult, ActionResult
import marl_factory_grid.modules.maintenance.constants as M
from marl_factory_grid.environment import constants as c
from marl_factory_grid.modules import Door, Machine
from marl_factory_grid.utils.results import TickResult, DoneResult, ActionResult
import marl_factory_grid.environment.constants as c
class Test(unittest.TestCase):
@ -14,6 +14,9 @@ class Test(unittest.TestCase):
return self.__class__.__name__
def __init__(self):
"""
Base test class for unit tests.
"""
super().__init__()
def __repr__(self):
@ -41,7 +44,11 @@ class Test(unittest.TestCase):
class MaintainerTest(Test):
def __init__(self):
"""
Tests whether the maintainer performs the correct actions and whether his actions register correctly in the env.
"""
super().__init__()
self.temp_state_dict = {}
pass
def tick_step(self, state) -> List[TickResult]:
@ -55,12 +62,12 @@ class MaintainerTest(Test):
if maintainer._closed_door_in_path(state):
self.assertEqual(maintainer.get_move_action(state).name, 'use_door')
elif maintainer._path:
elif maintainer._path and len(maintainer._path) > 1:
# can move
# print(maintainer.move(maintainer._path[1], state))
self.assertTrue(maintainer.move(maintainer._path[1], state))
print(maintainer.move(maintainer._path[1], state))
# self.assertTrue(maintainer.move(maintainer._path[1], state))
if not maintainer._path:
if maintainer._next and not maintainer._path:
# finds valid targets when at target location
route = maintainer.calculate_route(maintainer._last[-1], state.floortile_graph)
if entities_at_target_location := [entity for entity in state.entities.by_pos(route[-1])]:
@ -70,35 +77,47 @@ class MaintainerTest(Test):
def tick_post_step(self, state) -> List[TickResult]:
for maintainer in state.entities[M.MAINTAINERS]:
if maintainer._path:
# if action was door use: was door opened successfully?
# was door opened successfully?
if maintainer._closed_door_in_path(state):
door = next(
(entity for entity in state.entities.by_pos(maintainer._path[0]) if isinstance(entity, Door)),
None)
self.assertEqual(door.is_open, True)
# self.assertEqual(door.is_open, True)
# when stepping off machine, did maintain action work?
return []
def on_check_done(self, state) -> List[DoneResult]:
for maintainer in state.entities[M.MAINTAINERS]:
temp_state = maintainer._status
self.temp_state_dict[maintainer.identifier] = temp_state
print(self.temp_state_dict)
return []
class DirtAgentTest(Test):
def __init__(self):
"""
Tests whether the dirt agent will perform the correct actions and whether the actions register correctly in the
environment.
"""
super().__init__()
pass
def on_init(self, state, lvl_map):
# dirtagent richtig gespawnt?
return []
def on_reset(self):
return []
def tick_step(self, state) -> List[TickResult]:
# check observation correct?
# can open doors
# can find way
# can move
# clean action success? action result valid
for agent in state.entities[c.AGENT]:
print(agent)
# has valid actionresult
self.assertIsInstance(agent.state, ActionResult)
self.assertEqual(agent.state.validity, True)
return []
def tick_post_step(self, state) -> List[TickResult]: