| """Tests for dvnets.environments.environment.""" |
|
|
| from absl.testing import absltest |
|
|
| from cliport import tasks |
| from cliport.environments import environment |
|
|
| ASSETS_PATH = 'dvnets/environments/assets/' |
|
|
|
|
| class EnvironmentTest(absltest.TestCase): |
|
|
| def test_environment_action(self): |
| env = environment.Environment(ASSETS_PATH) |
| task = tasks.BlockInsertion() |
| env.set_task(task) |
| env.seed(0) |
| agent = task.oracle(env) |
| obs = env.reset() |
| info = None |
| done = False |
| for _ in range(10): |
| act = agent.act(obs, info) |
| self.assertTrue(env.action_space.contains(act)) |
| obs, _, done, info = env.step(act) |
| if done: |
| break |
|
|
|
|
| if __name__ == '__main__': |
| absltest.main() |
|
|
