finished documenmtation of steptests and test

marl_factory_grid/environment/tests.py

@@ -17,7 +17,8 @@ class Test(unittest.TestCase):
 
     def __init__(self):
         """
-        Base test class for unit tests.
+        Base test class for unit tests that provides base functions to be overwritten that are automatically called by
+        the StepTests class.
         """
         super().__init__()
 

marl_factory_grid/utils/states.py

@@ -309,6 +309,9 @@ class Gamestate(object):
 
 class StepTests:
     def __init__(self, *args):
+        """
+        The StepTests class is responsible for calling all tests and their respective hooks at the right time.
+        """
         if args:
             self.tests = list(args)
         else:
@@ -326,6 +329,12 @@ class StepTests:
         return True
 
     def do_all_init(self, state, lvl_map) -> bool:
+        """
+        Iterate all **Tests** that override the *on_check_done* hook.
+
+        :return: valid
+        :rtype: bool
+        """
         for test in self.tests:
             if test_init_printline := test.on_init(state, lvl_map):
                 state.print(test_init_printline)
@@ -335,10 +344,6 @@ class StepTests:
         """
         Iterate all **Tests** that override the *tick_step* hook.
 
-        :return:    List of Results
-        """        """
-        Iterate all **Tests** that override the *on_check_done* hook.
-
         :return: List of Results
         """
         test_results = list()