def agent_in_little_city(self, with_blue=False): red_agents = Composite("My reds") test_agent = create_resident(RED_AGENTS, TEST_ANUM) red_agents += test_agent blue_agents = Composite("My blues") if with_blue: for i in range(0, SMALL_GRID * SMALL_GRID - 1): blue_agents += create_resident(BLUE_AGENTS, TEST_ANUM + 1) my_city = Env("Small city for test", width=SMALL_GRID, height=SMALL_GRID, members=[red_agents, blue_agents]) return (test_agent, my_city)
def test_create_agent(self): """ Test that creating an agent works. """ fred = create_resident(RED_AGENTS, TEST_ANUM) freds_nm = group_names[RED_TEAM] + str(TEST_ANUM) self.assertEqual(freds_nm, str(fred))
def test_other_group_index(self): red_agent = create_resident(RED_AGENTS, TEST_ANUM) self.assertEqual(BLUE_TEAM, other_group_index(red_agent)) blue_agent = create_resident(BLUE_AGENTS, TEST_ANUM) self.assertEqual(RED_TEAM, other_group_index(blue_agent))