def test_championship_location(self): # No home_cmp, no updated team = Team(id="frc7332", team_number=7332) self.assertIsNone(team.home_cmp) self.assertIsNone(team.updated) self.assertIsNone(team.championship_location) # home_cmp, no updated team.home_cmp = 'cmp' self.assertIsNotNone(team.home_cmp) self.assertIsNone(team.updated) self.assertIsNone(team.championship_location) # no home_cmp, updated team.home_cmp = None team.updated = datetime.now() self.assertIsNone(team.home_cmp) self.assertIsNotNone(team.updated) self.assertIsNone(team.championship_location) # home_cmp, updated, no event team.home_cmp = 'cmp' self.assertIsNotNone(team.home_cmp) self.assertIsNotNone(team.updated) self.assertIsNone(team.championship_location) event = Event(id="{}{}".format(team.updated.year, team.home_cmp), event_type_enum=EventType.CMP_FINALS, event_short="cmp", year=team.updated.year) event.put() # home_cmp, updated, event, no city self.assertIsNotNone(team.home_cmp) self.assertIsNotNone(team.updated) self.assertIsNone(team.championship_location) # home_cmp, updated, event, city event.city = 'Atlanta' self.assertIsNotNone(team.home_cmp) self.assertIsNotNone(team.updated) self.assertEqual(team.championship_location, {team.updated.year: 'Atlanta'})