def get_current_state_test(): w = World(5, 5, [(1, 1, 3), (2, 2, 4)], [(3, 3, 3), (4, 4, 4)], -1, 13, 13) a = Agent(0, 0) assert(get_current_state(w, a) == (0, 0, False, True, True, True, True)) a._set_position(3, 3) w.pick_up(1, 1) w.pick_up(1, 1) w.pick_up(1, 1) assert(get_current_state(w, a) == (3, 3, False, False, True, True, True))
def p_random_test(): agent = Agent(3, 3) world = World(5, 5, [(1, 1, 3)], [(1, 2, 3)], -1, 13, 13) q = QTable(world) assert (p_random(agent, world, q) in ["north", "south", "east", "west"]) world = World(5, 5, [(4, 3, 3)], [(1, 1, 3)], -1, 13, 13) assert (p_random(agent, world, q) == "east") agent.pick_up() agent._set_position(1, 2) assert (p_random(agent, world, q) == "north")