def test_alternative_decorator_doc(): def my_func(): """ Foo bar """ pass decorated = alternative(my_func) assert decorated.__doc__.strip() == "Foo bar"
def test_alternative_decorator(): mock = Mock() mock.return_value = 'baz' a_alternative = alternative(mock) a1 = a_alternative('foo') assert isinstance(a1, Alternative) assert a1.func() == 'baz' assert mock.mock_calls == [call('foo')]