def test_init_copies_team_list(self):
teams = [self.team_a, self.team_b]
test = TargetFinder(self.map, teams)
teams[0] = 'not a team'
teams[1] = 'nope'
for unit in self.a_units + self.b_units:
test_team = test.get_team(unit)
self.assertTrue(test_team.is_on_team(unit))
def test_get_team(self):
teams = [self.team_a, self.team_b]
test = TargetFinder(self.map, teams)
for unit in self.a_units + self.b_units:
test_team = test.get_team(unit)
self.assertTrue(test_team.is_on_team(unit))