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