Skip to content
Snippets Groups Projects
Commit b515a1c4 authored by spiglerg's avatar spiglerg
Browse files

fixed tests

parent 25bb98a0
No related branches found
No related tags found
No related merge requests found
......@@ -3,6 +3,7 @@
from flatland.core.env import RailEnv
from flatland.core.transitions import Grid4Transitions
from flatland.core.transitionmap import GridTransitionMap
import numpy as np
"""Tests for `flatland` package."""
......@@ -44,7 +45,9 @@ def test_rail_environment_single_agent():
north_west_turn]],
dtype=np.uint16)
rail_env = RailEnv(rail_map, number_of_agents=1)
rail = GridTransitionMap(width=3, height=3, transitions=transitions)
rail.grid = rail_map
rail_env = RailEnv(rail, number_of_agents=1)
for _ in range(200):
_ = rail_env.reset()
......
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