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)
