def test_add_all(self):
locator = TorusLocator(2, 5)
locator.add_all([DummyAgent(), DummyAgent()])
self.assertEqual(len(locator.get_empty_slots()), 8)
def test_add_too_many_agents(self):
locator = TorusLocator(5, 5)
self.assertEqual(25, locator.add_all([DummyAgent() for _ in range(30)]))
