def test_agent_is_at_treasure(self): world = GridWorld(10, 10) world.add_agent(Agent(0, 0)) world.add_object(Treasure(8, 8)) self.assertFalse(world.agent.is_at_any_object(world.get_objects_by_type(Treasure))) world.agent.x = 8 world.agent.y = 8 self.assertTrue(world.agent.is_at_any_object(world.get_objects_by_type(Treasure)))
def test_add_treasure(self): world = GridWorld(10, 10) world.add_object(Treasure(8, 8)) self.assertEqual(1, len(world.get_objects())) self.assertListEqual([Treasure], world.get_object_types()) self.assertEqual(1, len(world.get_objects_by_type(Treasure)))
def test_add_agent(self): world = GridWorld(10, 10) world.add_agent(Agent(0, 0)) self.assertEqual(0, world.agent.x) self.assertEqual(0, world.agent.y) self.assertEqual(1, len(world.get_objects())) self.assertListEqual([Agent], world.get_object_types()) self.assertEqual(1, len(world.get_objects_by_type(Agent)))