def test_already_at_goal(self): me = MapEnvironment(self.map) agent = LRTAStarAgent(OnlineSearchProblem(MapActionFunction(self.map), MapGoalTestFunction("A"), MapStepCostFunction(self.map)), MapPerceptToStateFunction(), self.HF()) me.add_new_agent(agent, "A") expected_actions = [NoOpAction()] me.add_environment_view(EnvironmentViewMock(expected_actions)) me.step_until_done()
def test_normal_search(self): me = MapEnvironment(self.map) agent = OnlineDFSAgent(OnlineSearchProblem(MapActionFunction(self.map), MapGoalTestFunction("G"), MapStepCostFunction(self.map)), MapPerceptToStateFunction()) me.add_new_agent(agent, "A") locations = ['C', 'A', 'B', 'A', 'B', 'E', 'B', 'D', 'B', 'D', 'G'] expected_actions = get_move_to_actions_array(locations) me.add_environment_view(EnvironmentViewMock(expected_actions)) me.step_until_done()
def test_no_path(self): me = MapEnvironment(self.map) agent = LRTAStarAgent(OnlineSearchProblem(MapActionFunction(self.map), MapGoalTestFunction("G"), MapStepCostFunction(self.map)), MapPerceptToStateFunction(), self.HF()) me.add_new_agent(agent, "A") locations = ['B', 'A', 'B', 'C', 'B', 'C', 'D', 'C', 'D', 'E', 'D', 'E', 'F', 'E'] expected_actions = get_move_to_actions_array(locations) expected_actions.pop() me.add_environment_view(EnvironmentViewMock(expected_actions)) me.step(14)
def test_already_at_goal(self): me = MapEnvironment(self.map) agent = OnlineDFSAgent( OnlineSearchProblem(MapActionFunction(self.map), MapGoalTestFunction("A"), MapStepCostFunction(self.map)), MapPerceptToStateFunction()) me.add_new_agent(agent, "A") expected_actions = [NoOpAction()] me.add_environment_view(EnvironmentViewMock(expected_actions)) me.step_until_done()
def test_no_path(self): map = ExtendableMap() map.add_bidirectional_link('A', 'B', 1) me = MapEnvironment(map) agent = OnlineDFSAgent(OnlineSearchProblem(MapActionFunction(map), MapGoalTestFunction("G"), MapStepCostFunction(map)), MapPerceptToStateFunction()) me.add_new_agent(agent, "A") locations = ['B', 'A', 'B', 'A'] expected_actions = get_move_to_actions_array(locations) me.add_environment_view(EnvironmentViewMock(expected_actions)) me.step_until_done()
def test_normal_search(self): me = MapEnvironment(self.map) agent = OnlineDFSAgent( OnlineSearchProblem(MapActionFunction(self.map), MapGoalTestFunction("G"), MapStepCostFunction(self.map)), MapPerceptToStateFunction()) me.add_new_agent(agent, "A") locations = ['C', 'A', 'B', 'A', 'B', 'E', 'B', 'D', 'B', 'D', 'G'] expected_actions = get_move_to_actions_array(locations) me.add_environment_view(EnvironmentViewMock(expected_actions)) me.step_until_done()
def test_no_path(self): map = ExtendableMap() map.add_bidirectional_link('A', 'B', 1) me = MapEnvironment(map) agent = OnlineDFSAgent( OnlineSearchProblem(MapActionFunction(map), MapGoalTestFunction("G"), MapStepCostFunction(map)), MapPerceptToStateFunction()) me.add_new_agent(agent, "A") locations = ['B', 'A', 'B', 'A'] expected_actions = get_move_to_actions_array(locations) me.add_environment_view(EnvironmentViewMock(expected_actions)) me.step_until_done()
def test_no_path(self): me = MapEnvironment(self.map) agent = LRTAStarAgent( OnlineSearchProblem(MapActionFunction(self.map), MapGoalTestFunction("G"), MapStepCostFunction(self.map)), MapPerceptToStateFunction(), self.HF()) me.add_new_agent(agent, "A") locations = [ 'B', 'A', 'B', 'C', 'B', 'C', 'D', 'C', 'D', 'E', 'D', 'E', 'F', 'E' ] expected_actions = get_move_to_actions_array(locations) expected_actions.pop() me.add_environment_view(EnvironmentViewMock(expected_actions)) me.step(14)