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)