Skip to content
Snippets Groups Projects
Commit f37e25b6 authored by Egli Adrian (IT-SCI-API-PFI)'s avatar Egli Adrian (IT-SCI-API-PFI)
Browse files

test changed

parent 372aedc4
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
env_data/tests/Level_distance_map_shortest_path.pkl 0 → 100644
+ 0
− 0
View file @ f37e25b6
File added
tests/test_flatland_envs_rail_env_shortest_paths.py
+ 92
− 21
View file @ f37e25b6
import sys
import numpy as np import numpy as np
from flatland.core.grid.grid4 import Grid4TransitionsEnum from flatland.core.grid.grid4 import Grid4TransitionsEnum
...@@ -195,28 +197,97 @@ def test_get_shortest_paths_max_depth(): ...@@ -195,28 +197,97 @@ def test_get_shortest_paths_max_depth():
"[{}] actual={},expected={}".format(agent_handle, actual[agent_handle], expected[agent_handle]) "[{}] actual={},expected={}".format(agent_handle, actual[agent_handle], expected[agent_handle])
def test_get_shortest_paths_agent_handle(): def test_get_shortest_paths_agent_handle():
env = load_flatland_environment_from_file('test_002.pkl', 'env_data.tests') env = load_flatland_environment_from_file('Level_distance_map_shortest_path.pkl', 'env_data.tests')
actual = get_shortest_paths(env.distance_map, max_depth=2,agent_handle=1) actual = get_shortest_paths(env.distance_map, agent_handle=1)
expected = {
1: [
WalkingElement(position=(3, 18), direction=3,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(3, 17), next_direction=3)),
WalkingElement(position=(3, 17), direction=3,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(3, 16), next_direction=3)),
]
}
for agent_handle in expected:
assert np.array_equal(actual[agent_handle], expected[agent_handle]), \
"[{}] actual={},expected={}".format(agent_handle, actual[agent_handle], expected[agent_handle])
print(actual, file=sys.stderr)
expected = {1: [
WalkingElement(position=(20, 5), direction=3,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(20, 4), next_direction=3)),
WalkingElement(position=(20, 4), direction=3,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(20, 3), next_direction=3)),
WalkingElement(position=(20, 3), direction=3,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(20, 2), next_direction=3)),
WalkingElement(position=(20, 2), direction=3,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(20, 1), next_direction=3)),
WalkingElement(position=(20, 1), direction=3,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(20, 0), next_direction=3)),
WalkingElement(position=(20, 0), direction=3,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(19, 0), next_direction=0)),
WalkingElement(position=(19, 0), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(18, 0), next_direction=0)),
WalkingElement(position=(18, 0), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(17, 0), next_direction=0)),
WalkingElement(position=(17, 0), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(16, 0), next_direction=0)),
WalkingElement(position=(16, 0), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(15, 0), next_direction=0)),
WalkingElement(position=(15, 0), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(15, 1), next_direction=1)),
WalkingElement(position=(15, 1), direction=1,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(15, 2), next_direction=1)),
WalkingElement(position=(15, 2), direction=1,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(15, 3), next_direction=1)),
WalkingElement(position=(15, 3), direction=1,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(15, 4), next_direction=1)),
WalkingElement(position=(15, 4), direction=1,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(15, 5), next_direction=1)),
WalkingElement(position=(15, 5), direction=1,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(14, 5), next_direction=0)),
WalkingElement(position=(14, 5), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(13, 5), next_direction=0)),
WalkingElement(position=(13, 5), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(12, 5), next_direction=0)),
WalkingElement(position=(12, 5), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(11, 5), next_direction=0)),
WalkingElement(position=(11, 5), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(10, 5), next_direction=0)),
WalkingElement(position=(10, 5), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(9, 5), next_direction=0)),
WalkingElement(position=(9, 5), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(8, 5), next_direction=0)),
WalkingElement(position=(8, 5), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_RIGHT,
next_position=(8, 6), next_direction=1)),
WalkingElement(position=(8, 6), direction=1,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(7, 6), next_direction=0)),
WalkingElement(position=(7, 6), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(6, 6), next_direction=0)),
WalkingElement(position=(6, 6), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.MOVE_FORWARD,
next_position=(5, 6), next_direction=0)),
WalkingElement(position=(5, 6), direction=0,
next_action_element=RailEnvNextAction(action=RailEnvActions.STOP_MOVING,
next_position=(5, 6), next_direction=0))
]}
if False:
for agent_handle in expected:
assert np.array_equal(actual[agent_handle], expected[agent_handle]), \
"[{}] actual={},expected={}".format(agent_handle, actual[agent_handle], expected[agent_handle])
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment