maintainer test

marl_factory_grid/environment/tests.py

@@ -1,17 +1,20 @@
-import abc
 from typing import List
+import unittest
 
-from marl_factory_grid.utils.results import TickResult, DoneResult
+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
 
 
-class Test(abc.ABC):
+class Test(unittest.TestCase):
 
     @property
     def name(self):
         return self.__class__.__name__
 
     def __init__(self):
-        pass
+        super().__init__()
 
     def __repr__(self):
         return f'{self.name}'
@@ -35,9 +38,77 @@ class Test(abc.ABC):
         return []
 
 
-class FirstTest(Test):
+class MaintainerTest(Test):
 
     def __init__(self):
-        print("firstTest")
         super().__init__()
         pass
+
+    def tick_step(self, state) -> List[TickResult]:
+        for maintainer in state.entities[M.MAINTAINERS]:
+
+            # has valid actionresult
+            self.assertIsInstance(maintainer.state, ActionResult)
+            self.assertEqual(maintainer.state.validity, True)
+
+            # will open doors when standing in front
+            if maintainer._closed_door_in_path(state):
+                self.assertEqual(maintainer.get_move_action(state).name, 'use_door')
+
+            if maintainer._path:
+                # can move
+                print(maintainer.move(maintainer._path[1], state))  # 0 immer false ausser schritt 1, 1 meistens true nicht immer
+                # self.assertTrue(maintainer.move(maintainer._path[1], state))
+
+            else:
+                # finds valid targets oder hier?
+                route = maintainer.calculate_route(maintainer._last[-1], state.flootile_graph)
+                if entities_at_target_location := [entity for entity in state.entities.by_pos(route[-1])]:
+                    self.assertTrue(any(isinstance(e, Machine) for e in entities_at_target_location))
+
+        return []
+
+    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?
+                if maintainer._closed_door_in_path(state):
+                    # print(maintainer.get_move_action(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)
+
+        return []
+
+
+class DirtAgentTest(Test):
+
+    def __init__(self):
+        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
+        return []
+
+    def tick_post_step(self, state) -> List[TickResult]:
+        # action success?
+        # collisions? if yes, reported?
+        return []
+
+    def on_check_done(self, state) -> List[DoneResult]:
+        return []
+
+# class ItemAgentTest(Test):