def test_remove_agent(self):
locator = TorusLocator(5, 6)
agent = DummyAgent()
locator.add_agent(agent, (0, 0))
self.assertNotIn((0, 0), locator.get_empty_slots())
locator.remove_agent(agent)
self.assertIn((0, 0), locator.get_empty_slots())
def test_get_empty_slots(self):
locator = TorusLocator(2, 5)
self.assertEqual(locator.get_empty_slots(), [(x, y) for x in range(2) for y in range(5)])
locator.add_agent(DummyAgent(), (0, 1))
self.assertFalse((0, 1) in locator.get_empty_slots())
def test_add_all(self):
locator = TorusLocator(2, 5)
locator.add_all([DummyAgent(), DummyAgent()])
self.assertEqual(len(locator.get_empty_slots()), 8)
