TESTS: Adding tests for robot as well as examine envs

Author: vikashplus

robohive/tests/test_all.py

@@ -7,6 +7,8 @@
 from robohive.tests.test_claws import TestClaws
 from robohive.tests.test_tcdm import TestTCDM
 
+from robohive.tests.test_examine_env import TestExamineEnv
+from robohive.tests.test_robot import TestRobot
 
 if __name__ == '__main__':
     unittest.main()

robohive/tests/test_examine_env.py

@@ -2,7 +2,9 @@
 import click.testing
 import unittest
 from robohive.utils.examine_env import main as examine_env
-class TestMain(unittest.TestCase):
+
+
+class TestExamineEnv(unittest.TestCase):
     def test_main(self):
         # Call your function and test its output/assertions
         print("Testing env with random policy")
@@ -21,5 +23,9 @@ def test_offscreen_rendering(self):
                                             "--num_episodes", 1, \
                                             "--render", "offscreen",\
                                             "--camera_name", "top_acam"])
-        # print(result.output.strip())
-        self.assertEqual(result.exception, None)
+        print(result.output.strip())
+        self.assertEqual(result.exception, None)
+
+
+if __name__ == '__main__':
+    unittest.main()

robohive/tests/test_robot.py

@@ -0,0 +1,11 @@
+import unittest
+from robohive.robot.robot import demo_robot
+
+class TestRobot(unittest.TestCase):
+    def test_robot(self):
+        # Call your function and test its output/assertions
+        print("Testing robot")
+        demo_robot()
+
+if __name__ == '__main__':
+    unittest.main()