Esempio n. 1
0
def test_inheritance_and_multiple_dispatch():
    @dispatch
    def f(x: A, y: A):
        return type(x), type(y)

    @dispatch
    def f(x: A, y: B):
        return 0

    assert f(A(), A()) == (A, A)
    assert f(A(), C()) == (A, C)
    assert f(A(), B()) == 0
    assert f(C(), B()) == 0
    assert raises(NotImplementedError, lambda: f(B(), B()))
def test_inheritance_and_multiple_dispatch():
    @dispatch
    def f(x: A, y: A):
        return type(x), type(y)

    @dispatch
    def f(x: A, y: B):
        return 0

    assert f(A(), A()) == (A, A)
    assert f(A(), C()) == (A, C)
    assert f(A(), B()) == 0
    assert f(C(), B()) == 0
    assert raises(NotImplementedError, lambda: f(B(), B()))
Esempio n. 3
0
def test_singledispatch():
    @dispatch
    def f(x: int):
        return x + 1

    @dispatch
    def g(x: int):
        return x + 2

    @dispatch
    def f(x: float):
        return x - 1

    assert f(1) == 2
    assert g(1) == 3
    assert f(1.0) == 0

    assert raises(NotImplementedError, lambda: f('hello'))
def test_singledispatch():
    @dispatch
    def f(x: int):
        return x + 1

    @dispatch
    def g(x: int):
        return x + 2

    @dispatch
    def f(x: float):
        return x - 1

    assert f(1) == 2
    assert g(1) == 3
    assert f(1.0) == 0

    assert raises(NotImplementedError, lambda: f('hello'))