unittest shows wrong line number for decorated methods

[james10424]

VS Code version

1.64.0-insider

Extension version

v2021.12.1559732655

OS type

macOS

OS version

11.6

Python distribution

python.org

Python version

3.7

Language server

Pylance

Expected behaviour

VSCode should show a "run" button next to my test methods, with decorator or not

Actual behaviour

VSCode shows the "run" button next to the decorator definition instead of the test method

Steps to reproduce

Enable unittest in VSCode

Paste this code

import unittest

def decorated(f):
    def wrapper(*args):
        print("decorated")
        return f(*args)
    return wrapper

class TestStuff(unittest.TestCase):
    @decorated
    def test_stuff(self):
        assert 1 == 1

You will see the "run" button appear next to def wrapper(*args):, but it should be on def test_stuff(self):

Edit:

Bonus: every decorated method will go to its decorated function, so you can have 100 tests all on that one line

Logs

> ~/.local/share/virtualenvs/[REDUCTED_PATH]-SM8362Ue/bin/python ~/.vscode-insiders/extensions/ms-python.python-2021.12.1559732655/pythonFiles/testing_tools/unittest_discovery.py [REDUCTED_PATH] test*.py
cwd: [REDUCTED_PATH]

Edit:
Output of that discovery:

start
test_lol:TestStuff:test_stuff:4

Code of Conduct

• I agree to follow this project's Code of Conduct

[brettcannon]

Unfortunately there's not much we can do here as we have to ask the method object where it exists and because you wrapped it the code object points to the decorator function and not the original method object.

I would also suggest you use the functools.wraps function to make sure your decorators look as much like the wrapped function as possible.

[james10424]

@brettcannon Thank you for the suggestion, that solved the problem!