def test_num_modules_on_schedule(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_on_schedule() == 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_on_schedule() == 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_on_schedule() == 5)
