def test_insert_2(self):
head = Node()
head.buff = [1, 2, 3, 4, 5]
head.count = 5
n1 = Node()
n1.buff = [4, 0, 0, 0, 0]
n1.count = 1
n2 = Node()
n2.buff = [5, 6, 7, 8, 9]
n2.count = 5
head.next = n1
n1.next = n2
sol = Solution()
sol.insert_at(head, 10, 11)
self.assertEqual(n2.buff, [5, 6, 7, 8, 11])
self.assertEqual(n2.next.buff, [9, 0, 0, 0, 0])
def test_insert_1(self):
head = Node()
head.buff = [1, 2, 3, 0, 0]
head.count = 3
n1 = Node()
n1.buff = [4, 0, 0, 0, 0]
n1.count = 1
n2 = Node()
n2.buff = [5, 6, 0, 0, 0]
n2.count = 2
head.next = n1
n1.next = n2
sol = Solution()
sol.insert_at(head, 0, 0)
self.assertEqual(head.buff, [0, 1, 2, 3, 0])
sol.insert_at(head, 0, 0)
self.assertEqual(head.buff, [0, 0, 1, 2, 3])
sol.insert_at(head, 0, 0)
self.assertEqual(head.buff, [0, 0, 0, 1, 2])
self.assertEqual(n1.buff, [3, 4, 0, 0, 0])
