def test_save_to_context(self): context = dict() def dont_extract(number): return number @extracts("arg") def extract_arg(number): return number @extracts("arg1", "arg2") def extract_arg1_and_arg2(num1, num2): return num1, num2 @extracts("args") def extract_args(num1, num2): return num1, num2 @extracts("from_callable") class CallableExtractor: def __call__(self, number): return number class Extractor: def method(self, number): return number @classmethod def class_method(cls, number): return number @staticmethod def static_method(number): return number wrapped_method = extracts("from_method")(Extractor().method) wrapped_class_method = extracts("from_class_method")(Extractor().class_method) wrapped_static_method = extracts("from_static_method")(Extractor().static_method) save_to_context(context, dont_extract)(-1) save_to_context(context, extract_arg)(0) save_to_context(context, extract_arg1_and_arg2)(1, 2) save_to_context(context, extract_args)(1, 2) save_to_context(context, CallableExtractor())(3) save_to_context(context, wrapped_method)(4) save_to_context(context, wrapped_class_method)(5) save_to_context(context, wrapped_static_method)(6) assert_that(context, is_(equal_to(dict( arg=0, arg1=1, arg2=2, args=(1, 2), from_callable=3, from_method=4, from_class_method=5, from_static_method=6, ))))
def test_chain_in_a_chain(self): chain = Chain( extracts("arg")(lambda: 20), Chain(lambda arg: arg * 10, ), ) assert_that( chain(), is_(equal_to(200)), )
def test_chain_functions_can_use_decorators(self): chain = Chain( extracts("arg")(lambda: 20), lambda arg: arg * 10, ) assert_that( chain(), is_(equal_to(200)), )
def test_chain_functions_can_use_decorators(self): chain = Chain( extracts("arg")(lambda: 20), lambda arg: arg * 10, ) assert_that( chain(), is_(equal_to(200)), )
def test_chain_in_a_chain(self): chain = Chain( extracts("arg")(lambda: 20), Chain( lambda arg: arg * 10, ), ) assert_that( chain(), is_(equal_to(200)), )
def test_overriding_extracts(self): context = dict() @extracts("arg") def extractarg(number): return number func = extracts("param")(extractarg) save_to_context(context, func)(10) assert_that(context, is_(equal_to(dict(param=10))))
def test_overriding_extracts(self): context = dict() @extracts("arg") def extractarg(number): return number func = extracts("param")(extractarg) save_to_context(context, func)(10) assert_that(context, is_(equal_to(dict(param=10))))