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