def test_get_first(self):
"""
Tests that the first node of a list is returned
"""
l = LinkedList()
l.insert_first(1)
self.assertEqual(l.get_first().data, 1)
l.insert_first(2)
self.assertEqual(l.get_first().data, 2)
def test_remove_first_size_3(self):
"""
Tests the removal of first node in list when list has three nodes
"""
l = LinkedList()
l.insert_first("c")
l.insert_first("b")
l.insert_first("a")
l.remove_first()
self.assertEqual(l.size(), 2)
self.assertEqual(l.get_first().data, "b")
l.remove_first()
self.assertEqual(l.size(), 1)
self.assertEqual(l.get_first().data, "c")
def test_insert_at_0_on_empty_list(self):
"""
Tests insertion of node on empty list
"""
l = LinkedList()
l.insert_at("hi", 0)
self.assertEqual(l.get_first().data, "hi")
def test_remove_first_size_1(self):
"""
Tests the removal of first node in list when list has one node
"""
l = LinkedList()
l.insert_first("a")
l.remove_first()
self.assertEqual(l.size(), 0)
self.assertEqual(l.get_first(), None)
