Skip to content
Snippets Groups Projects
Commit dde6d690 authored by hagrid67's avatar hagrid67
Browse files

added test to load msgpack env and add agent_static

parent d80453c2
No related branches found
No related tags found
No related merge requests found
from flatland.envs.rail_env import RailEnv
# from flatland.envs.agent_utils import EnvAgent
from flatland.envs.agent_utils import EnvAgentStatic
def test_load_env():
env = RailEnv(10, 10)
env.load("env-data/tests/test-10x10.mpk")
agent_static = EnvAgentStatic((0, 0), 2, (5, 5))
env.add_agent_static(agent_static)
assert env.get_num_agents() == 1
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment