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_init(self): world = GridWorld(10, 10) self.assertEqual(0, world.step) self.assertEqual(10, world.width) self.assertEqual(10, world.height) self.assertIsNone(world.agent) self.assertEqual(0, len(world.get_objects())) self.assertEqual(0, len(world.get_object_types()))
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)))