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