def test_add_challenger(self): m1 = Challenger("Kakouak", "lol") m2 = Challenger("Kakouik", "lol") m3 = Challenger("Kikouak", "lol") m4 = Challenger("Kikouik", "lol") g = Group([m1, m2], "Test adding challengers") g.give_point(m1, 203) g.give_point(m2, 22) g.add_challenger(m4, 19) g.sort_challengers_by_points() self.assertSequenceEqual(g.challengers, [m1, m2, m4]) g.add_challenger(m3, 184) g.sort_challengers_by_points() self.assertSequenceEqual(g.challengers, [m1, m3, m2, m4])
def test_remove_challenger(self): m1 = Challenger("Kakouak", "lol") m2 = Challenger("Kakouik", "lol") m3 = Challenger("Kikouak", "lol") m4 = Challenger("Kikouik", "lol") g = Group([m1, m2, m3, m4], "Test removing challengers") g.give_point(m1, 203) g.give_point(m2, 22) g.give_point(m3, 13) g.remove_challenger(m4) g.sort_challengers_by_points() self.assertSetEqual(set(g.challengers), {m1, m2, m3}) g.remove_challenger(m3) g.sort_challengers_by_points() self.assertSetEqual(set(g.challengers), {m1, m2})