def test_remove_dead_agents(self):
locator = TorusLocator(5, 6)
a1 = DummyAgent()
a2 = DummyAgent()
locator.add_agent(a1, (0, 0))
locator.add_agent(a2, (0, 1))
self.assertEqual(a2, locator.get_neighbour(a1))
a2.dead = True
self.assertIsNone(locator.get_neighbour(a1))
def test_neighbourhood(self):
locator = TorusLocator(5, 6)
a1 = DummyAgent()
a2 = DummyAgent()
a3 = DummyAgent()
locator.add_agent(a1, (0, 0))
locator.add_agent(a2, (0, 1))
locator.add_agent(a3, (0, 3))
self.assertEqual(a2, locator.get_neighbour(a1))
self.assertFalse(a3 == locator.get_neighbour(a1))
self.assertIsNone(locator.get_neighbour(a3))
