added test hooks (like rules)

marl_factory_grid/environment/tests.py

@@ -38,5 +38,6 @@ class Test(abc.ABC):
 class FirstTest(Test):
 
     def __init__(self):
+        print("firstTest")
         super().__init__()
         pass

marl_factory_grid/testing/test_config.yaml

@@ -22,26 +22,41 @@ Agents:
     - Inventory
     - DropOffLocations
     - Maintainers
+    # This is special for agents, as each one is different and can act as an adversary e.g.
+    Positions:
+      - (16, 7)
+      - (16, 6)
+      - (16, 3)
+      - (16, 4)
+      - (16, 5)
 Entities:
   Batteries:
     initial_charge: 0.8
     per_action_costs: 0.02
-  ChargePods: {}
+  ChargePods:
-  Destinations: {}
+    coords_or_quantity: 2
+  Destinations:
+    coords_or_quantity: 1
+    spawn_mode: GROUPED
   DirtPiles:
+    coords_or_quantity: 10
+    initial_amount: 2
     clean_amount: 1
     dirt_spawn_r_var: 0.1
-    initial_amount: 2
-    initial_dirt_ratio: 0.05
     max_global_amount: 20
     max_local_amount: 5
-  Doors: {}
+  Doors:
-  DropOffLocations: {}
+  DropOffLocations:
+    coords_or_quantity: 1
+    max_dropoff_storage_size: 0
   GlobalPositions: {}
   Inventories: {}
-  Items: {}
+  Items:
-  Machines: {}
+    coords_or_quantity: 5
-  Maintainers: {}
+  Machines:
+    coords_or_quantity: 2
+  Maintainers:
+    coords_or_quantity: 1
   Zones: {}
 
 General:
@@ -49,33 +64,34 @@ General:
   individual_rewards: true
   level_name: large
   pomdp_r: 3
-  verbose: false
+  verbose: True
-  tests: true
+  tests: false
 
 Rules:
-  SpawnAgents: {}
+  # Environment Dynamics
-  DoneAtBatteryDischarge: {}
-  Collision:
-    done_at_collisions: false
-  AssignGlobalPositions: {}
-  DoneAtDestinationReachAny: {}
-  DestinationReachReward: {}
-  SpawnDestinations:
-    n_dests: 1
-    spawn_mode: GROUPED
-  DoneOnAllDirtCleaned: {}
-  SpawnDirt:
-    spawn_freq: 15
   EntitiesSmearDirtOnMove:
     smear_ratio: 0.2
   DoorAutoClose:
     close_frequency: 10
-  ItemRules:
+  MoveMaintainers:
-    max_dropoff_storage_size: 0
+
-    n_items: 5
+  # Respawn Stuff
-    n_locations: 5
+  RespawnDirt:
-    spawn_frequency: 15
+    respawn_freq: 15
-  MaxStepsReached:
+  RespawnItems:
-    max_steps: 10
+    respawn_freq: 15
-#  AgentSingleZonePlacement:
+
-#    n_zones: 4
+  # Utilities
+  WatchCollisions:
+    done_at_collisions: false
+
+  # Done Conditions
+  DoneAtDestinationReachAny:
+  DoneOnAllDirtCleaned:
+  DoneAtBatteryDischarge:
+  DoneAtMaintainerCollision:
+  DoneAtMaxStepsReached:
+    max_steps: 500
+
+Tests:
+  FirstTest: {}

marl_factory_grid/testing/test_run.py

@@ -7,7 +7,7 @@ from marl_factory_grid.environment.factory import Factory
 
 if __name__ == '__main__':
     # Render at each step?
-    render = True
+    render = False
 
     # Path to config File
     path = Path('test_config.yaml')

marl_factory_grid/utils/config_parser.py

@@ -161,8 +161,8 @@ class FactoryConfigParser(object):
 
         return self._load_smth(rules, Rule)
 
-    def load_env_tests(self) -> List[Rule]:
+    def load_env_tests(self) -> List[Test]:
-        return self._load_smth(self.tests, None)  # Test
+        return self._load_smth(self.tests, Test)
 
     def _load_smth(self, config, class_obj):
         rules = list()