def test_remove_at_out_of_bounds(self): """ Tests for return of None when called on index that is out of bounds """ l = LinkedList() l.insert_first("a") l.remove_at(1) self.assertEqual(l.remove_at(1), None)
def test_remove_at_given_index(self): """ Tests for removal of node at given index """ l = LinkedList() l.insert_last(1) l.insert_last(2) l.insert_last(3) l.insert_last(4) self.assertEqual(l.get_at(1).data, 2) l.remove_at(1) self.assertEqual(l.get_at(1).data, 3)
def test_remove_at_first_node(self): """ Tests removal of first node """ l = LinkedList() l.insert_last(1) l.insert_last(2) l.insert_last(3) l.insert_last(4) self.assertEqual(l.get_at(0).data, 1) l.remove_at(0) self.assertEqual(l.get_at(0).data, 2)