You are probably mixing together decorator taking args and not taking args. You would be right, if contextmanager would be called as a function, like this:

def some_decorator(some_arg=None):
    def inner(func):
        def wrapper(*args, **kwargs):
            ...
            return func(*args, **kwargs)
        return wrapper
    return inner

@some_decorator()  # note the parentheses
def foo(x):
    print(x)

foo('aaa')

However, you have to take args and kwargs in the wrapper if you use the decorator without args:

def some_decorator(func):
    def wrapper(*args, **kwargs):
        ...
        return func(*args, **kwargs)
    return wrapper

@some_decorator  # no parentheses
def foo(x):
    print(x)

foo('aaa')

Fixed, thanks.

You welcome :)

You welcome :)

Sure. It's in the description as well.

Thanks!

I added a note, that python 2 is used. Thanks!

Good idea, fixed, thanks!

Because I write "production" code. I ended with printing inputs, but it's more convinient when the test prints the information itself. Many katas do this.

Python

You could describe what was tested (inputs). If you get "basic tests - 44 should equal 43" you have no clue what is really wrong.

Some example tests would be great.

Little bit more tests would be great.

More tests and better description of tests would be great.

Thanks, added.