diff --git a/examples/pybullet/gym/pybullet_envs/examples/testEnv.py b/examples/pybullet/gym/pybullet_envs/examples/testEnv.py new file mode 100644 index 000000000..e43131cb8 --- /dev/null +++ b/examples/pybullet/gym/pybullet_envs/examples/testEnv.py @@ -0,0 +1,45 @@ +import os +import inspect +currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))) +parentdir = os.path.dirname(os.path.dirname(currentdir)) +os.sys.path.insert(0,parentdir) +import pybullet_envs +import gym +import argparse + + +def test(args): + + env = gym.make(args.env) + env.env.configure(args) + if (args.render): + env.render(mode="human") + env.reset() + print("action space:") + sample = env.action_space.sample() + action = sample*0.0 + print("action=") + print(action) + for i in range(args.steps): + obs,rewards,done,_ =env.step(action) + print("obs=") + print(obs) + print("rewards") + print (rewards) + print ("done") + print(done) + + +def main(): + import argparse + parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter) + parser.add_argument('--env', help='environment ID', default='AntBulletEnv-v0') + parser.add_argument('--seed', help='RNG seed', type=int, default=0) + parser.add_argument('--render', help='Render in OpenGL Window', type=str, default=0) + parser.add_argument('--steps', help='Number of steps', type=int, default=1) + parser.add_argument('--bla',help='bla',type=int, default=42) + args = parser.parse_args() + test(args) + +if __name__ == '__main__': + main()