test adjacent obs

rlcube/rlcube/envs/__init__.py

@@ -3,6 +3,16 @@ from .cube2 import Cube2Env
 __all__ = ["Cube2Env"]
 
 if __name__ == "__main__":
+    print("Testing Cube2Env.from_obs")
     env = Cube2Env()
     env2 = Cube2Env.from_obs(env.obs())
-    print(env2.state)
+    print(env2.state, env2.is_solved())
+    env2.print_js_code()
+    print()
+
+    print("Testing Cube2Env.adjacent_obs")
+    env = Cube2Env()
+    adjacent_obs = env.adjacent_obs()
+    for i in range(12):
+        env = Cube2Env.from_obs(adjacent_obs[i])
+        env.print_js_code()

rlcube/rlcube/envs/cube2.py

@@ -252,3 +252,6 @@ class Cube2Env(gym.Env):
             if np.mean(self.state[i]) != self.state[i][0]:
                 return False
         return True
+
+    def print_js_code(self):
+        print(f"rotationController.setState({self.state.tolist()});")