Improve tests

Signed-off-by: guillemdb <[email protected]>

src/plangym/videogames/nes.py

@@ -66,11 +66,11 @@ def __setattr__(self, name, value):
         if name == "_wrapped":
             super().__setattr__(name, value)
         else:
-            setattr(self._wrapped, name, value)
+            setattr(self._wrapped, name, value)  # pragma: no cover
 
     def __delattr__(self, name):
         """Delete an attribute from the wrapped object."""
-        delattr(self._wrapped, name)
+        delattr(self._wrapped, name)  # pragma: no cover
 
     def step(
         self, action: ActType

tests/videogames/test_nes.py

@@ -21,5 +21,16 @@ def env(request):
     return request.param()
 
 
-# class TestMarioEnv(TestPlangymEnv):
-#    pass
+class TestMarioEnv:
+    def test_get_keys_to_action(self, env):
+        vals = env.gym_env.get_keys_to_action()
+        assert isinstance(vals, dict)
+
+    def test_get_action_meanings(self, env):
+        vals = env.gym_env.get_action_meanings()
+        assert isinstance(vals, list)
+
+    def test_buttons(self, env):
+        buttons = env.gym_env.buttons()
+        assert isinstance(buttons, list), buttons
+        assert all(isinstance(b, str) for b in buttons)