tests and organising

Author: zbeucler2018

HotWheelsEnv.py

@@ -114,6 +114,44 @@ def discrete_to_multibinary(self, action):
         return arr.astype(np.uint8)
 
 
+class SingleActionEnv(Discretizer):
+    """
+    Restricts the agent's actions to a a single button per action
+            
+            []
+            , ['B']
+            , ['A']
+            , ['UP']
+            , ['DOWN']
+            , ['LEFT']
+            , ['RIGHT']
+            , ['L', 'R']
+    """
+
+    def __init__(self, env):
+        super().__init__(env=env, 
+                        buttons=env.unwrapped.buttons, 
+                        combos=[  
+            []
+            , ['B']
+            , ['A']
+            , ['UP']
+            , ['DOWN']
+            , ['LEFT']
+            , ['RIGHT']
+            , ['L', 'R']
+        ])
+
+        self.original_env = env
+    
+    def get_discrete_button_meaning(self, action):
+        """
+        get button from discrete action
+        """
+        multibinary_action = self.discrete_to_multibinary(action)
+        return self.original_env.get_action_meaning(multibinary_action)        
+
+
 class FixSpeed(gym.Wrapper):
     """
     Fixes env bug so the speed is accurate
@@ -153,44 +191,6 @@ def step(self, action):
         return observation, reward, terminated, truncated, info
 
 
-class SingleActionEnv(Discretizer):
-    """
-    Restricts the agent's actions to a a single button per action
-            
-            []
-            , ['B']
-            , ['A']
-            , ['UP']
-            , ['DOWN']
-            , ['LEFT']
-            , ['RIGHT']
-            , ['L', 'R']
-    """
-
-    def __init__(self, env):
-        super().__init__(env=env, 
-                        buttons=env.unwrapped.buttons, 
-                        combos=[  
-            []
-            , ['B']
-            , ['A']
-            , ['UP']
-            , ['DOWN']
-            , ['LEFT']
-            , ['RIGHT']
-            , ['L', 'R']
-        ])
-
-        self.original_env = env
-    
-    def get_discrete_button_meaning(self, action):
-        """
-        get button from discrete action
-        """
-        multibinary_action = self.discrete_to_multibinary(action)
-        return self.original_env.get_action_meaning(multibinary_action)
-
-
 class TerminateOnCrash(gym.Wrapper):
     """
     A wrapper that ends the episode if the mean of the observation is above a certain threshold

test_gym_wrappers.py

@@ -0,0 +1,112 @@
+import unittest
+
+import retro
+
+
+from HotWheelsEnv import GameStates, FixSpeed, DoTricks, TerminateOnCrash, NorrmalizeBoost, SingleActionEnv
+
+
+
+
+
+class TestGameStates(unittest.TestCase):
+    """ Tests if a env can be created with each GameState """
+
+    def tearDown(self):
+        self.env.close()
+        self.env = None
+
+    def test_dino_single(self):
+        self.env = retro.make(
+            game="HotWheelsStuntTrackChallenge-gba", 
+            render_mode="rgb_array", 
+            state=GameStates.SINGLE.value
+        )
+
+    def test_dino_single_points(self):
+        self.env = retro.make(
+            game="HotWheelsStuntTrackChallenge-gba", 
+            render_mode="rgb_array", 
+            state=GameStates.SINGLE_POINTS.value
+        )
+
+    @unittest.skip(f"Skip until retro can take different data.json filenames")
+    def test_dino_multi(self):
+        self.env = retro.make(
+            game="HotWheelsStuntTrackChallenge-gba", 
+            render_mode="rgb_array", 
+            state=GameStates.MULTIPLAYER.value
+        )
+
+
+from retro import Actions
+
+
+class TestWrappers(unittest.TestCase):
+
+    def setUp(self):
+        self.env = retro.make(
+            game="HotWheelsStuntTrackChallenge-gba", 
+            render_mode="rgb_array", 
+            state=GameStates.SINGLE.value
+        )
+        _, _ = self.env.reset(seed=42)
+
+    def tearDown(self):
+        self.env.close()
+        self.env = None
+
+    def test_FixSpeed(self):
+        self.env = FixSpeed(self.env)
+        random_action = self.env.action_space.sample()
+        observation, reward, terminated, truncated, info = self.env.step(random_action)
+
+    def test_DoTricks(self):
+        self.env = DoTricks(self.env)
+        random_action = self.env.action_space.sample()
+        observation, reward, terminated, truncated, info = self.env.step(random_action)
+
+    def test_TerminateOnCrash(self):
+        self.env = TerminateOnCrash(self.env)
+        random_action = self.env.action_space.sample()
+        observation, reward, terminated, truncated, info = self.env.step(random_action)
+
+    @unittest.skip('the change to data.json isnt working (doesnt detect boost entry)')
+    def test_NorrmalizeBoost(self):
+        self.env = NorrmalizeBoost(self.env)
+        random_action = self.env.action_space.sample()
+        observation, reward, terminated, truncated, info = self.env.step(random_action)
+
+    @unittest.skip('broken')
+    def test_SingleActionEnv(self):
+        self.env = SingleActionEnv(self.env)
+        random_action = self.env.action_space.sample()
+        observation, reward, terminated, truncated, info = self.env.step(random_action)
+
+    def test_huh(self):
+        self.env.close()
+        self.env = None
+        self.env = retro.make(
+            game="HotWheelsStuntTrackChallenge-gba", 
+            render_mode="rgb_array", 
+            state=GameStates.SINGLE.value,
+            use_restricted_actions=Actions.DISCRETE
+        )
+        _, _ = self.env.reset(seed=42)
+
+        random_action = self.env.action_space.sample()
+
+        # print(random_action)
+        # print(self.env.get_action_meaning(random_action))
+        # print(self.env.action_to_array(random_action))
+
+
+    
+
+
+
+
+
+
+if __name__ == '__main__':
+    unittest.main()