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)))