Início / coding / Perguntas / 77977601
Accepted
mike rodent
Asked: 2024-02-12 00:58:21 +0800 CST

Como posso testar o que meu arquivo `__main__` faz?

  • 772

Estou usando o pytest. Nota W10.
Estarei executando meu aplicativo indo > python src/core. "core" é meu módulo central, com um __main__.pyarquivo.

Por esse motivo estou colocando meu diretório "testes" em [raiz do projeto]/src/core.

Eu tenho um arquivo de teste test_xxx.py. Nele tento importar __main__:

import __main__

... 

def test_xxxxx():
   print(f'main {__main__} type {type(__main__)}')

Isso imprime o seguinte:

main <module '__main__' from 'D:\\apps\\Python\\virtual_envs\\doc_indexer v2.0.0\\Scripts\\pytest.exe\\__main__.py'> type <class 'module'>

... em outras palavras, ele não está importando meu local __main__.pydo módulo "principal" local, mas o próprio pytest __main__de seu módulo. Até agora tudo compreensível. Então eu olho para sys.path:

path: D:\My Documents\software projects\EclipseWorkspace\doc_indexer\src\core\tests
path: D:\My Documents\software projects\EclipseWorkspace\doc_indexer\src\core
path: D:\apps\Python\virtual_envs\doc_indexer v2.0.0\Scripts\pytest.exe
path: D:\apps\Python\PyQt5
...

Agora estou coçando a cabeça, não pela primeira vez com o pytest. Este caminho ...src\core está listado antes de ...Scripts\pytest.exe. Por que o pytest identificaria o seu próprio __main__.pye o importaria, antes de importar __main__.pyde um caminho que o precede?

Eu também tentei vários experimentos com importlib. Nada parece funcionar.

Então, geralmente, existe alguma maneira de realmente importar esse arquivo ...src/core/ __main__py?

Sei que esse arquivo deve fazer coisas mínimas antes de passar para outro arquivo, mas para completar, gostaria apenas de testar se main()de fato faz isso. Existe alguma maneira de executar __main__.main()um teste?

python

1 respostas

  1. Best Answer

    Todo programa Python cria um módulo nomeado __main__na inicialização, com base no script que está sendo executado.

    Como resultado, import __main__ sempre encontra um módulo com esse nome, o que significa que não é necessário defini-lo a partir, digamos, de um arquivo chamado __main__.py. No seu caso, você executou pytest(que é principalmente um wrapper simples em torno de import pytest; pytest.console_main(). (O script pytest, armazenado em um diretório em seu caminho de pesquisa, é separado de um módulo definido por pytest.pyem sua biblioteca Python.)

    Uma solução seria garantir que você possa importar __main__.pyusando outro pacote, de forma que somepackage.__main__seja diferente do __main__.

    No entanto, __main__.pynunca se destina a ser importado como um módulo. É o arquivo que define um script quando você tenta executar um diretório ou um arquivo compactado como um script. Se contiver código que precisa ser testado, eu recomendaria fatorá-lo em um módulo separado com __main__.pyas próprias importações.

    • 1

relate perguntas