Esempio n. 1
0
def test_solution_0():
    l1 = add_two_numbers.ListNode(1)
    l2 = add_two_numbers.ListNode(2)
    s = add_two_numbers.Solution()
    res = s.addTwoNumbers(l1, l2)

    assert res.val == 3
    assert res.next == None
Esempio n. 2
0
def test_solution_2():
    l1 = add_two_numbers.ListNode(5)
    l2 = add_two_numbers.ListNode(2)
    l3 = add_two_numbers.ListNode(7)
    l4 = add_two_numbers.ListNode(4)

    l1.next = l2
    l3.next = l4

    s = add_two_numbers.Solution()
    res = s.addTwoNumbers(l1, l3)

    assert res.val == 2
    assert res.next.val == 7
    assert res.next.next == None
def test_add_two_numbers():
    n1 = add_two_numbers.ListNode(2)
    n2 = add_two_numbers.ListNode(4)
    n3 = add_two_numbers.ListNode(3)
    n1.next = n2
    n2.next = n3

    n4 = add_two_numbers.ListNode(5)
    n5 = add_two_numbers.ListNode(6)
    n6 = add_two_numbers.ListNode(4)
    n4.next = n5
    n5.next = n6

    n7 = add_two_numbers.ListNode(7)
    n8 = add_two_numbers.ListNode(0)
    n9 = add_two_numbers.ListNode(8)
    n7.next = n8
    n8.next = n9

    assert add_two_numbers.Solution().add_two_numbers(n1, n4).val == n7.val
    assert add_two_numbers.Solution().add_two_numbers(
        n1, n4).next.val == n7.next.val
    assert add_two_numbers.Solution().add_two_numbers(
        n1, n4).next.next.val == n7.next.next.val
Esempio n. 4
0
def test_solution_4():
    l1 = add_two_numbers.ListNode(5)
    l2 = add_two_numbers.ListNode(7)

    l3 = add_two_numbers.ListNode(7)
    l4 = add_two_numbers.ListNode(4)
    l5 = add_two_numbers.ListNode(4)
    l6 = add_two_numbers.ListNode(4)

    l1.next = l2

    l3.next = l4
    l4.next = l5
    l5.next = l6

    s = add_two_numbers.Solution()
    res = s.addTwoNumbers(l1, l3)

    assert res.val == 2
    assert res.next.val == 2
    assert res.next.next.val == 5
    assert res.next.next.next.val == 4
    assert res.next.next.next.next == None
    def from_list(cls, to_be_linked: list) -> add_two_numbers.ListNode:
        if not to_be_linked:
            return None

        return add_two_numbers.ListNode(to_be_linked[0],
                                        cls.from_list(to_be_linked[1:]))