mirror of https://github.com/flaggo/pydu.git
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
|
from pydu.inspect import (get_func_full_args, func_accepts_var_args,
|
||
|
|
func_accepts_kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
class Person:
|
||
|
|
def no_arguments(self):
|
||
|
|
return None
|
||
|
|
|
||
|
|
def one_argument(self, something):
|
||
|
|
return something
|
||
|
|
|
||
|
|
def just_args(self, *args):
|
||
|
|
return args
|
||
|
|
|
||
|
|
def just_kwargs(self, **kwargs):
|
||
|
|
return kwargs
|
||
|
|
|
||
|
|
def all_kinds(self, name, address='home', age=25, *args, **kwargs):
|
||
|
|
return kwargs
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_func_full_args():
|
||
|
|
# no arguments
|
||
|
|
assert get_func_full_args(Person.no_arguments) == []
|
||
|
|
# one argument
|
||
|
|
assert get_func_full_args(Person.one_argument) == [('something',)]
|
||
|
|
# all_arguments
|
||
|
|
arguments = [('name',), ('address', 'home'), ('age', 25), ('*args',), ('**kwargs',)]
|
||
|
|
assert get_func_full_args(Person.all_kinds) == arguments
|
||
|
|
|
||
|
|
|
||
|
|
def test_func_accepts_var_args():
|
||
|
|
# has args
|
||
|
|
assert func_accepts_var_args(Person.just_args)
|
||
|
|
# no args
|
||
|
|
assert not func_accepts_var_args(Person.one_argument)
|
||
|
|
|
||
|
|
|
||
|
|
def test_func_accepts_kwargs():
|
||
|
|
# has kwargs
|
||
|
|
assert func_accepts_kwargs(Person.just_kwargs)
|
||
|
|
# no kwargs
|
||
|
|
assert not func_accepts_kwargs(Person.one_argument)
|