Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import sys
- import functools
- def takes(*types):
- def decorator(func):
- @functools.wraps(func)
- def wrapper(*args):
- for i, (arg, expected_type) in enumerate(zip(args, types)):
- if not isinstance(arg, expected_type):
- raise TypeError(f"Argument {i+1} has wrong type: expected {expected_type}, got {type(arg)}")
- return func(*args)
- return wrapper
- return decorator
- exec(sys.stdin.read())
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement