def test_override_arg(self):
override_decorator = Decorators.decorator(self.override_x)
override_g = override_decorator(self.g)
assert override_g(0) == 75
override_g3 = override_decorator(self.g3)
assert override_g3(25, 0) == 100
def test_decorator_lambda(self):
"""test that a decorator is called, and properly calls the underlying function"""
l = lambda x: x + 21
assert l(3) == 24
chatty_decorator = Decorators.decorator(self.chatty)
chatty_l = chatty_decorator(l)
assert chatty_l(3) == 24
def test_decorator(self):
"""test that a decorator is called, and properly calls the underlying function"""
chatty_decorator = Decorators.decorator(self.chatty)
chatty_g = chatty_decorator(self.g)
result = chatty_g(3)
assert result == 28
assert self.g.calls[-1] == "Calling 'g'"
def test_decorator(self):
"""test that a decorator is called, and properly calls the underlying function"""
chatty_decorator = Decorators.decorator(self.chatty)
chatty_g = chatty_decorator(self.g)
result = chatty_g(3)
assert result == 28
assert self.g.calls[-1] == "Calling 'g'"
def test_decorator_lambda(self):
"""test that a decorator is called, and properly calls the underlying function"""
l = lambda x: x + 21
assert l(3) == 24
chatty_decorator = Decorators.decorator(self.chatty)
chatty_l = chatty_decorator(l)
assert chatty_l(3) == 24
def test_decorator_signature(self):
"""tests that a decorated function retains the signature of the underlying function"""
chatty_decorator = Decorators.decorator(self.chatty)
chatty_f = chatty_decorator(self.f)
info = Decorators.decorator_helpers.getinfo(self.f)
chatty_info = Decorators.decorator_helpers.getinfo(chatty_f)
assert info == chatty_info
assert info.__doc__ == chatty_info.__doc__
def test_decorator_signature(self):
"""tests that a decorated function retains the signature of the underlying function"""
chatty_decorator = Decorators.decorator(self.chatty)
chatty_f = chatty_decorator(self.f)
info = Decorators.decorator_helpers.getinfo(self.f)
chatty_info = Decorators.decorator_helpers.getinfo(chatty_f)
assert info == chatty_info
assert info.__doc__ == chatty_info.__doc__
def test_decorator_lambda_calling_frame_extendedarg(self):
"""tests that a decorated function can get the frame of the calling function when a lambda is being decorated, and pass it as an extended argument"""
l = lambda x, calling_frame: x + 21
frames_so_far = len(getattr(l, "call_frames", []))
callf_decorator = Decorators.decorator(self.callfdec2, [("calling_frame", None)], calling_frame_arg="calling_frame")
callf_l = callf_decorator(l)
assert callf_l(100, "nonsense") == 100 + 21
assert len(l.call_frames) == frames_so_far + 1
call_frame = l.call_frames[-1]
assert call_frame
# check that the argument given for calling frame is overridden
assert call_frame != "nonsense"
assert call_frame.f_code.co_name == "test_decorator_lambda_calling_frame_extendedarg"
def test_decorator_calling_frame_arg(self):
"""tests that a decorated function can get the frame of the calling function"""
frames_so_far = len(getattr(self.g, "call_frames", []))
callf_decorator = Decorators.decorator(self.callfdec, ['y', ('z', 3)], calling_frame_arg="calling_frame")
callf_g = callf_decorator(self.g)
assert callf_g(100, 4) == 100 + 12 + 25
assert len(self.g.call_frames) == frames_so_far + 1
call_frame = self.g.call_frames[-1]
assert call_frame
assert call_frame.f_code.co_name == "test_decorator_calling_frame_arg"
# check that calling_frame can't be overridden
some_frame = Decorators.inspect.currentframe()
raises(TypeError, callf_g, 100, 4, calling_frame=some_frame)
raises(TypeError, callf_g, 100, 4, some_frame)
def test_extend_decorator_signature(self):
"""tests that a decorated function can extend the signature of the underlying function"""
ext_decorator = Decorators.decorator(self.extdec, ['y', ('z', 3)])
ext_g = ext_decorator(self.g)
info = Decorators.decorator_helpers.getinfo(self.g)
ext_info = Decorators.decorator_helpers.getinfo(ext_g)
assert ext_info.__doc__ == info.__doc__
assert ext_info["name"] == info["name"]
assert ext_info["argnames"] == ['x', 'y', 'z']
assert ext_info["defarg"] == (3, )
assert ext_info["shortsign"] == 'x, y, z'
assert ext_info["fullsign"] == 'x, y, z=defarg[0]'
assert ext_info["arg0"] == 'x'
assert ext_info["arg1"] == 'y'
assert ext_info["arg2"] == 'z'
def test_decorator_calling_frame_arg(self):
"""tests that a decorated function can get the frame of the calling function"""
frames_so_far = len(getattr(self.g, "call_frames", []))
callf_decorator = Decorators.decorator(
self.callfdec, ['y', ('z', 3)], calling_frame_arg="calling_frame")
callf_g = callf_decorator(self.g)
assert callf_g(100, 4) == 100 + 12 + 25
assert len(self.g.call_frames) == frames_so_far + 1
call_frame = self.g.call_frames[-1]
assert call_frame
assert call_frame.f_code.co_name == "test_decorator_calling_frame_arg"
# check that calling_frame can't be overridden
some_frame = Decorators.inspect.currentframe()
raises(TypeError, callf_g, 100, 4, calling_frame=some_frame)
raises(TypeError, callf_g, 100, 4, some_frame)
def test_decorator_lambda_calling_frame_arg(self):
"""tests that a decorated function can get the frame of the calling function when a lambda is being decorated"""
l = lambda x: x + 21
frames_so_far = len(getattr(l, "call_frames", []))
callf_decorator = Decorators.decorator(self.callfdec, ['y', ('z', 3)], calling_frame_arg="calling_frame")
callf_l = callf_decorator(l)
assert callf_l(100, 4) == 100 + 12 + 21
assert len(l.call_frames) == frames_so_far + 1
call_frame = l.call_frames[-1]
assert call_frame
assert call_frame.f_code.co_name == "test_decorator_lambda_calling_frame_arg"
# check that calling_frame can't be overridden
some_frame = Decorators.inspect.currentframe()
raises(TypeError, callf_l, 100, 4, calling_frame=some_frame)
raises(TypeError, callf_l, 100, 4, some_frame)
def test_extend_decorator_signature(self):
"""tests that a decorated function can extend the signature of the underlying function"""
ext_decorator = Decorators.decorator(self.extdec, ['y', ('z', 3)])
ext_g = ext_decorator(self.g)
info = Decorators.decorator_helpers.getinfo(self.g)
ext_info = Decorators.decorator_helpers.getinfo(ext_g)
assert ext_info.__doc__ == info.__doc__
assert ext_info["name"] == info["name"]
assert ext_info["argnames"] == ['x', 'y', 'z']
assert ext_info["defarg"] == (3,)
assert ext_info["shortsign"] == 'x, y, z'
assert ext_info["fullsign"] == 'x, y, z=defarg[0]'
assert ext_info["arg0"] == 'x'
assert ext_info["arg1"] == 'y'
assert ext_info["arg2"] == 'z'
def test_decorator_lambda_calling_frame_extendedarg(self):
"""tests that a decorated function can get the frame of the calling function when a lambda is being decorated, and pass it as an extended argument"""
l = lambda x, calling_frame: x + 21
frames_so_far = len(getattr(l, "call_frames", []))
callf_decorator = Decorators.decorator(
self.callfdec2, [("calling_frame", None)],
calling_frame_arg="calling_frame")
callf_l = callf_decorator(l)
assert callf_l(100, "nonsense") == 100 + 21
assert len(l.call_frames) == frames_so_far + 1
call_frame = l.call_frames[-1]
assert call_frame
# check that the argument given for calling frame is overridden
assert call_frame != "nonsense"
assert call_frame.f_code.co_name == "test_decorator_lambda_calling_frame_extendedarg"
def test_decorator_lambda_calling_frame_arg(self):
"""tests that a decorated function can get the frame of the calling function when a lambda is being decorated"""
l = lambda x: x + 21
frames_so_far = len(getattr(l, "call_frames", []))
callf_decorator = Decorators.decorator(
self.callfdec, ['y', ('z', 3)], calling_frame_arg="calling_frame")
callf_l = callf_decorator(l)
assert callf_l(100, 4) == 100 + 12 + 21
assert len(l.call_frames) == frames_so_far + 1
call_frame = l.call_frames[-1]
assert call_frame
assert call_frame.f_code.co_name == "test_decorator_lambda_calling_frame_arg"
# check that calling_frame can't be overridden
some_frame = Decorators.inspect.currentframe()
raises(TypeError, callf_l, 100, 4, calling_frame=some_frame)
raises(TypeError, callf_l, 100, 4, some_frame)
def test_extend_decorator_existing(self):
"""tests that a decorated function can extend the underlying function, handling existing arguments right"""
ext_decorator = Decorators.decorator(self.extdec2, [('y', 5), ('z', 3), ('x', 0)])
ext_g = ext_decorator(self.g2)
info = Decorators.decorator_helpers.getinfo(self.g2)
ext_info = Decorators.decorator_helpers.getinfo(ext_g)
assert ext_info.__doc__ == info.__doc__
assert ext_info["name"] == info["name"]
assert ext_info["argnames"] == ['x', 'z', 'y']
assert ext_info["defarg"] == (0, 3, 5)
assert ext_info["shortsign"] == 'x, z, y'
assert ext_info["fullsign"] == 'x=defarg[0], z=defarg[1], y=defarg[2]'
assert ext_info["arg0"] == 'x'
assert ext_info["arg1"] == 'z'
assert ext_info["arg2"] == 'y'
result = ext_g()
assert result == (0 + 3*5) + 3 + 25
assert self.g2.calls[-1] == "Called with x=0, y=5, z=3"
def test_decorator_calling_frame_extendedarg(self):
"""tests that a decorated function can get the frame of the calling function and pass it as an extended argument"""
frames_so_far = len(getattr(self.g, "call_frames", []))
callf_decorator = Decorators.decorator(self.callfdec, ['y', ('z', 3), ("calling_frame", None)], calling_frame_arg="calling_frame")
callf_g = callf_decorator(self.g)
assert callf_g(100, 4) == 100 + 12 + 25
assert len(self.g.call_frames) == frames_so_far + 1
call_frame = self.g.call_frames[-1]
assert call_frame
assert call_frame.f_code.co_name == "test_decorator_calling_frame_extendedarg"
# check that calling_frame can be overridden, but the result is still correct
some_frame = Decorators.inspect.currentframe().f_back
assert callf_g(100, 4, calling_frame=some_frame) == 100 + 12 + 25
assert len(self.g.call_frames) == frames_so_far + 2
assert self.g.call_frames[-1] != some_frame
assert callf_g(100, 4, 5, some_frame) == 100 + 20 + 25
assert len(self.g.call_frames) == frames_so_far + 3
assert self.g.call_frames[-1] != some_frame
def test_extend_decorator_existing(self):
"""tests that a decorated function can extend the underlying function, handling existing arguments right"""
ext_decorator = Decorators.decorator(self.extdec2, [('y', 5), ('z', 3),
('x', 0)])
ext_g = ext_decorator(self.g2)
info = Decorators.decorator_helpers.getinfo(self.g2)
ext_info = Decorators.decorator_helpers.getinfo(ext_g)
assert ext_info.__doc__ == info.__doc__
assert ext_info["name"] == info["name"]
assert ext_info["argnames"] == ['x', 'z', 'y']
assert ext_info["defarg"] == (0, 3, 5)
assert ext_info["shortsign"] == 'x, z, y'
assert ext_info["fullsign"] == 'x=defarg[0], z=defarg[1], y=defarg[2]'
assert ext_info["arg0"] == 'x'
assert ext_info["arg1"] == 'z'
assert ext_info["arg2"] == 'y'
result = ext_g()
assert result == (0 + 3 * 5) + 3 + 25
assert self.g2.calls[-1] == "Called with x=0, y=5, z=3"
def test_decorator_calling_frame_extendedarg(self):
"""tests that a decorated function can get the frame of the calling function and pass it as an extended argument"""
frames_so_far = len(getattr(self.g, "call_frames", []))
callf_decorator = Decorators.decorator(
self.callfdec, ['y', ('z', 3), ("calling_frame", None)],
calling_frame_arg="calling_frame")
callf_g = callf_decorator(self.g)
assert callf_g(100, 4) == 100 + 12 + 25
assert len(self.g.call_frames) == frames_so_far + 1
call_frame = self.g.call_frames[-1]
assert call_frame
assert call_frame.f_code.co_name == "test_decorator_calling_frame_extendedarg"
# check that calling_frame can be overridden, but the result is still correct
some_frame = Decorators.inspect.currentframe().f_back
assert callf_g(100, 4, calling_frame=some_frame) == 100 + 12 + 25
assert len(self.g.call_frames) == frames_so_far + 2
assert self.g.call_frames[-1] != some_frame
assert callf_g(100, 4, 5, some_frame) == 100 + 20 + 25
assert len(self.g.call_frames) == frames_so_far + 3
assert self.g.call_frames[-1] != some_frame
def test_extend_decorator(self):
"""tests that a decorated function can extend the underlying function"""
ext_decorator = Decorators.decorator(self.extdec, ['y', ('z', 3)])
ext_g = ext_decorator(self.g)
assert ext_g(100, 4) == 100 + 12 + 25
assert self.g.calls[-1] == "Called with x=100, y=4, z=3"
def test_extend_decorator(self):
"""tests that a decorated function can extend the underlying function"""
ext_decorator = Decorators.decorator(self.extdec, ['y', ('z', 3)])
ext_g = ext_decorator(self.g)
assert ext_g(100, 4) == 100 + 12 + 25
assert self.g.calls[-1] == "Called with x=100, y=4, z=3"
