TESTS: Updating tests to check everything for different versions. Add…

…ing test for stable baselines

robohive/tests/test_sb.py

@@ -0,0 +1,11 @@
+from stable_baselines3 import PPO
+import robohive
+from robohive.utils import gym
+
+from robohive import robohive_arm_suite
+for env_name in sorted(robohive_arm_suite):
+    print(f"Training {env_name} ========================================")
+    env = gym.make(env_name)
+    model = PPO("MlpPolicy", env, verbose=0)
+    model.learn(total_timesteps=2)
+    break

robohive/tests/test_versions.sh

@@ -1,23 +1,23 @@
 pip uninstall -y gym
 pip uninstall -y gymnasium
+pip uninstall -y stable-baselines3
 
 echo "=================== Testing gym==0.13 ==================="
 pip install gym==0.13
-python tests/test_arms.py
-python tests/test_examine_env.py
-python tests/test_examine_robot.py
-python tests/test_logger.py
-python tests/test_robot.py
+python tests/test_all.py
 pip uninstall -y gym
 
 echo "=================== Testing gym==0.26.2 ==================="
 pip install gym==0.26.2
-python tests/test_arms.py
 python tests/test_all.py
 pip uninstall -y gym
 
 echo "=================== Testing gymnasium ==================="
 pip install gymnasium
-python tests/test_arms.py
 python tests/test_all.py
+
+echo "=================== Testing Stable Baselines ==================="
+pip install stable-baselines3
+python tests/test_sb.py
 pip uninstall -y gymnasium
+pip uninstall -y stable-baselines3