def test2(self):
head = Node()
head.buff = [1, 2, 3]
head.count = 3
n1 = Node()
n1.buff = [4]
n1.count = 1
n2 = Node()
n2.buff = [5, 6]
n2.count = 2
head.next = n1
n1.next = n2
sol = Solution()
res = sol.get_idx_at(head, 0)
exp = 1
self.assertEqual(res, exp)
res = sol.get_idx_at(head, 1)
exp = 2
self.assertEqual(res, exp)
res = sol.get_idx_at(head, 4)
exp = 5
self.assertEqual(res, exp)
res = sol.get_idx_at(head, 5)
exp = 6
self.assertEqual(res, exp)
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 test1(self):
head = Node()
head.buff = [1, 2, 3]
head.count = 3
n1 = Node()
n1.buff = [4, 5, 6]
n1.count = 3
head.next = n1
sol = Solution()
res = sol.get_idx_at(head, 0)
exp = 1
self.assertEqual(res, exp)
res = sol.get_idx_at(head, 1)
exp = 2
self.assertEqual(res, exp)
res = sol.get_idx_at(head, 4)
exp = 5
self.assertEqual(res, exp)
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])
