def test_remove_item_for_nonexistent_index(self): l = SortedList() l.insert_item_fast(1) l.insert_item_fast(2) l.insert_item_fast(3) with self.assertRaises(IndexError): l.remove_item(3)
def test_remove_item(self): l = SortedList() for i in range(10): l.insert_item_slow(i) l.remove_item(5) self.assertEqual(l.print_sorted_list(), [0, 1, 2, 3, 4, 6, 7, 8, 9])