def get_divisions_in_schedule(schedule): divisions = schedule.divisions.all() if not divisions.count(): # if there aren't divisions make one import common division = schedule.divisions.create(name='Created Automatically') for team in common.find_teams_for(schedule.season): division.teams.add(team) divisions = [division] return divisions
def test_find_teams_for(self): self.assertEquals(self.team, common.find_teams_for(self.season_old)[0]) self.assertEquals(self.team, common.find_teams_for(self.season_open)[0])