def test_add_team(self):
location = Location('Dublin', "culture1", 30)
team1 = Team('test_team1', 10)
self.assertTrue(location.add_team(team1))
team2 = Team('test_team2', 20)
self.assertTrue(location.add_team(team2))
team3 = Team('test_team3', 1)
self.assertFalse(location.add_team(team3))
self.assertTrue(len(location.teams) == 2)
def test_num_modules(self):
location = Location('Dublin', "culture1", 30)
location.add_team(Team('test_team1', 1))
location.teams[0].module = Module('test_module1', 600)
self.assertTrue(location.num_modules() == 1)
location.add_team(Team('test_team2', 1))
location.teams[1].module = Module('test_module2', 400)
location.add_team(Team('test_team3', 1))
location.teams[2].module = Module('test_module3', 200)
self.assertTrue(location.num_modules() == 3)
location.add_team(Team('test_team4', 1))
location.teams[3].module = Module('test_module4', 600)
location.add_team(Team('test_team5', 1))
location.teams[4].module = Module('test_module5', 800)
self.assertTrue(location.num_modules() == 5)
def test_total_module_progress(self):
location = Location('Dublin', "culture1", 30)
location.add_team(Team('test_team1', 1))
location.add_team(Team('test_team2', 1))
location.add_team(Team('test_team3', 1))
location.add_team(Team('test_team4', 1))
location.add_team(Team('test_team5', 1))
self.assertTrue(location.total_module_progress() == 0)
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)
