def test_build_list(self):
list_str = "1->2->3"
h1 = build_list(list_str)
self.assertEqual(list_str, entire_list(h1))
h2 = build_list([1, 2, 3])
self.assertEqual(list_str, entire_list(h2))
def test_add_two_numbers(self):
s = Solution()
l1 = build_list([2, 4, 3])
l2 = build_list([5, 6, 4, 8])
res = s.addTwoNumbers(l1, l2)
print(entire_list(res))
l1 = build_list([6, 1])
l2 = build_list([4])
res = s.addTwoNumbers(l1, l2)
print(entire_list(res))
def test_merge_klists(self):
lists_str = [
"1->4->5",
"1->3->4",
"2->6",
]
lists = [build_list(s) for s in lists_str]
s = Solution()
res = s.mergeKLists(lists)
self.assertEqual("1->1->2->3->4->4->5->6", entire_list(res))
def test_merge_two_lists(self):
h1 = build_list("1->2->4")
h2 = build_list("1->3->4")
s = Solution()
res = s.mergeTwoLists(h1, h2)
self.assertEqual("1->1->2->3->4->4", entire_list(res))
def test_remove_nth_from_end(self):
head = build_list([1, 2, 3, 4, 5])
s = Solution()
h = s.removeNthFromEnd(head, 2)
self.assertEqual(entire_list(h), "1->2->3->5")
def test_delete_node(self):
s = Solution()
head = build_list([4, 5, 1, 9])
print('before delete', entire_list(head))
h = s.deleteNode(head, 5)
print('after delete', entire_list(h))
def test_get_kth_from_end(self):
head = build_list([1, 2, 3, 4, 5])
s = Solution()
res = s.getKthFromEnd(head, 2)
self.assertEqual(entire_list(res), "4->5")
def test_entire_list(self):
node = ListNode(1, ListNode(2, ListNode(3)))
self.assertEqual("1->2->3", entire_list(node))