def test_remove_all(self): cl = CourseList() cl.insert(Course(1000)) for _ in range(20): cl.insert(Course(1200)) cl.insert(Course(1800)) self.assertEqual(cl.size(), 22) cl.remove_all(1200) self.assertEqual(cl.size(), 2)
def test_remove_all(): cl = CourseList() cl.insert(Course(1000, "", 0.0, 0.0)) for _ in range(20): cl.insert(Course(1200, "", 0.0, 0.0)) cl.insert(Course(1800, "", 0.0, 0.0)) assert cl.size() == 22 cl.remove_all(1200) assert cl.size() == 2