def test_base_wrapper(self):
obj_with_params = BaseWrapper(a=10, b='message')
obj_without_params = BaseWrapper()
def foo_fn():
return 'bar'
obj_with_fn = BaseWrapper(foo_fn)
self.assertEqual(obj_with_params.target_params, {'a': 10, 'b': 'message'})
self.assertEqual(obj_without_params.target_params, {})
self.assertEqual(obj_with_fn.target_fn(), 'bar')
def test_base_wrapper(self):
obj_with_params = BaseWrapper(a=10, b='message')
obj_without_params = BaseWrapper()
def foo_fn():
return 'bar'
obj_with_fn = BaseWrapper(foo_fn)
self.assertEqual(obj_with_params.target_params, {'a': 10, 'b': 'message'})
self.assertEqual(obj_without_params.target_params, {})
self.assertEqual(obj_with_fn.target_fn(), 'bar')
def test_filter_params(self):
# right arguments are kept and wrong dismmissed
def baz_fn(a, b, c=True):
if c:
return a + b
else:
return a
obj_ = BaseWrapper(baz_fn, a=10, b=100, d=1000)
self.assertEqual(obj_.filter_params(baz_fn), {'a': 10, 'b': 100})
# target_params is overriden using 'override' argument
self.assertEqual(obj_.filter_params(baz_fn, override={'c': False}),
{'a': 10, 'b': 100, 'c': False})
def test_set_params(self):
class Pipo():
def __init__(self):
self.name = 'pipo'
def __call__(self, message):
return message
pipo = Pipo()
obj = BaseWrapper(pipo)
# argument is set accordingly
obj.set_params(message='OK')
self.assertEqual(obj.target_params, {'message': 'OK'})
self.assertEqual(obj.target_fn(**obj.target_params), 'OK')
# invalid argument is passed
try:
obj = BaseWrapper(Pipo())
obj.set_params(invalid='KO')
except Exception:
self.assertEqual(obj.target_params, {})
def test_set_params(self):
class Pipo():
def __init__(self):
self.name = 'pipo'
def __call__(self, message):
return message
pipo = Pipo()
obj = BaseWrapper(pipo)
# argument is set accordingly
obj.set_params(message='OK')
self.assertEqual(obj.target_params, {'message': 'OK'})
self.assertEqual(obj.target_fn(**obj.target_params), 'OK')
# invalid argument is passed
try:
obj = BaseWrapper(Pipo())
obj.set_params(invalid='KO')
except Exception:
self.assertEqual(obj.target_params, {})
def test__check_params(self):
def bar_fn(a):
return str(a)
class Pipo():
def __init__(self):
self.name = 'pipo'
def __call__(self, message):
return message
pipo = Pipo()
obj_with_valid_fn = BaseWrapper(bar_fn, a=10, b='message')
obj_with_valid_callable_fn = BaseWrapper(pipo, c=10, d='message')
obj_with_invalid_fn = BaseWrapper([1, 2, 3], fn_name='invalid')
# target_fn is not a callable or function/method
with self.assertRaises(AttributeError):
obj_with_invalid_fn._check_params('fn_name')
# parameters is not in target_fn args
with self.assertRaises(ValueError):
obj_with_valid_fn._check_params(['c'])
obj_with_valid_callable_fn._check_params(['e'])
# params is in target_fn args
try:
obj_with_valid_fn._check_params(['a'])
obj_with_valid_callable_fn._check_params(['message'])
except Exception:
self.fail("_check_params() raised an unexpected exception")
# params is not a dict or list
with self.assertRaises(TypeError):
obj_with_valid_fn._check_params(None)
with self.assertRaises(TypeError):
obj_with_valid_fn._check_params('param_name')
def test__check_params(self):
def bar_fn(a):
return str(a)
class Pipo():
def __init__(self):
self.name = 'pipo'
def __call__(self, message):
return message
pipo = Pipo()
obj_with_valid_fn = BaseWrapper(bar_fn, a=10, b='message')
obj_with_valid_callable_fn = BaseWrapper(pipo, c=10, d='message')
obj_with_invalid_fn = BaseWrapper([1, 2, 3], fn_name='invalid')
# target_fn is not a callable or function/method
with self.assertRaises(AttributeError):
obj_with_invalid_fn._check_params('fn_name')
# parameters is not in target_fn args
with self.assertRaises(ValueError):
obj_with_valid_fn._check_params(['c'])
obj_with_valid_callable_fn._check_params(['e'])
# params is in target_fn args
try:
obj_with_valid_fn._check_params(['a'])
obj_with_valid_callable_fn._check_params(['message'])
except Exception:
self.fail("_check_params() raised an unexpected exception")
# params is not a dict or list
with self.assertRaises(TypeError):
obj_with_valid_fn._check_params(None)
with self.assertRaises(TypeError):
obj_with_valid_fn._check_params('param_name')