-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_wrapper.py
44 lines (34 loc) · 1.05 KB
/
test_wrapper.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import time
from simulationsandbox.utils.envs import ENVS
from simulationsandbox.simulator_wrapper import SimulationWrapper
MAX_AGENTS = 10
GRID_SIZE = 20
NUM_STEPS = 50
VIZUALIZE = True
SLEEP_TIME = 2
STEP_DELAY = 0.01
PRINT_DATA = False
SEED = 0
def test_wrapper():
for Env in ENVS.values():
env = Env(max_agents=MAX_AGENTS, grid_size=GRID_SIZE)
state = env.init_state(seed=SEED)
assert env.max_agents == MAX_AGENTS
assert env.grid_size == GRID_SIZE
assert state.agents.pos.shape[0] == MAX_AGENTS
sim = SimulationWrapper(env, state, seed=SEED, step_delay=STEP_DELAY, print_data=PRINT_DATA)
print('Started')
sim.start()
time.sleep(SLEEP_TIME)
assert sim.running is True
sim.pause()
print('Paused')
time.sleep(SLEEP_TIME)
assert sim.paused is True
print('Resumed')
sim.resume()
time.sleep(SLEEP_TIME)
assert sim.running is True
sim.stop()
print('stopped')
assert sim.stop_requested is True