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)
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")