def test_get_last(self): """ Tests that the last node of a list is returned """ l = LinkedList() l.insert_first(2) self.assertEqual(l.get_last().data, 2) self.assertEqual(l.get_last().next, None) l.insert_first(1) self.assertEqual(l.get_last().data, 2) self.assertEqual(l.get_last().next, None)
def test_insert_last(self): """ Tests adding a node to the end of a list """ l = LinkedList() l.insert_first("a") l.insert_last("b") self.assertEqual(l.size(), 2) self.assertEqual(l.get_last().data, "b")
def test_remove_last_3(self): """ Tests the removal of last node in a list with three nodes """ l = LinkedList() l.insert_first("c") l.insert_first("b") l.insert_first("a") l.remove_last() self.assertEqual(l.size(), 2) self.assertEqual(l.get_last().data, "b")
def test_remove_last_2(self): """ Tests the removal of last node in a list with two nodes """ l = LinkedList() l.insert_first("b") l.insert_first("a") l.remove_last() self.assertEqual(l.size(), 1) self.assertEqual(l.get_last().data, "a") self.assertEqual(l.get_last().next, None)