returns.assert_trace doesn't work with @safe-wrapped functions

First of all, thank you! I really appreciate your work and your help on the Telegram group 😃

Bug report

What’s wrong

returns.assert_trace fails to “intercept” containers created from @safe decorators.

# works
def func_1(n) -> Result[int, Exception]:
    if n == 0:
        return Failure(Exception())
    return Success(n + 1)

# doesn't work
@safe
def func_1(n) -> int:
    if n == 0:
        raise Exception()
    return n + 1


def test_func_1(returns: ReturnsAsserts):
    with returns.assert_trace(Success, func_1):
        flow(1, func_1)

The test with the wrapped function returns

FAILED returns_test.py::test_func_1 - Failed: No container Success was created

I tried doing this, but no difference:

def test_func_1(returns: ReturnsAsserts):
    with returns.assert_trace(Success, safe(func_1)):
        flow(1, safe(func_1))

How is that should be

The above test should pass

System information

• python version: 3.8.0
• returns version: 0.15.0
• pytest version (if any): 6.2.2

[ftruzzi]

Thanks! Really appreciate your quick reply and help.

I am really excited to see your lib! 👍

Here it is, I know it can be controversial but hopefully you’re not against it 🙃 https://github.com/ftruzzi/libgen_uploader

[sobolevn]

I am really excited to see your lib! 👍