def test_num_teams(self): location = Location('Dublin', "culture1", 30) self.assertTrue(location.num_teams() == 0) location.add_team(Team('test_team1', 1)) location.add_team(Team('test_team2', 1)) self.assertTrue(location.num_teams() == 2) location.add_team(Team('test_team3', 1)) self.assertTrue(location.num_teams() == 3) location.add_team(Team('test_team4', 1)) location.add_team(Team('test_team5', 1)) self.assertTrue(location.num_teams() == 5)