имя pytest не определено

Question

Я следую этому руководству по настройке pytest с моим проектом. Я создаю новый проект со следующей структурой и кодом:

/src
    /main.py

/tests
     /test_pytest.py

main.py

    def main():

    # Bunch of stuff

       print("End.")

# Entry point of the program
if (__name__ == '__main__'): 
    main()

test_pytest.py

import src.main as main

def test_main():
    assert main.main() == 4

Очевидно, что утверждение unittest терпят неудачу, но это не имеет значения. Visual Studio обнаруживает этот тест, но говорит следующее:

NameError: name 'main' is not definedpytest(./tests/test_pytest.py::test_main)

Я не понимаю, почему я не могу назвать свой модульный тест? Кажется, не имеет значения, какое имя я использую.

Answer 1

Можете попробовать:

from src import main

def test_main():
    assert main.main() == 4