Python Function.takes_arguments Examples

Example #1

File: test_types.py Project: poletaevvlad/CubeLang

def test_function_arguments_generic(args: typing.List[Type],
                                    return_type: Type):
    func = Function(([T, List(T)], Set(T)), ([Set(List(T))], Integer),
                    ([Set(T), List(T)], T))
    assert func.takes_arguments(args) == return_type

Example #2

File: test_types.py Project: poletaevvlad/CubeLang

def test_function_arguments(args: typing.List[Type], return_type: Type):
    func = Function(([Integer, Integer], Integer), ([Integer, Real], Bool),
                    ([Real, Real], Color), ([Real, List(Integer)], Real),
                    ([Color, Real, ...], Side))
    res = func.takes_arguments(args)
    assert res == return_type