Início / coding / Perguntas / 79181238
Accepted
OrenIshShalom
Asked: 2024-11-12 21:08:26 +0800 CST

Chamando seletivamente funções de teste com capsys

  • 772

Posso executar seletivamente test_function_1, substituindo a instrumentação dos meus conftest.pyacessórios 1

def test_function_1(instrumentation: dict[str, float]) -> None:
    assert instrumentation['a'] > instrumentation['b']

def test_function_2(capsys) -> None:
    print("Hello, pytest!")
    captured = capsys.readouterr()
    assert captured.out == "Hello, pytest!\n"

quando tento chamar test_function_2, não sei como passar capsyspara ele 2 :

import tests
import pytest # <--- doesn't help ...

def test_callee_with_instrumentation():
    tests.test_function_1({'a': 110, 'b': 55, 'g': 6000})

def test_callee_with_capsys():
    # tests.test_function_2()              # <--- TypeError: test_function_2() missing 1 required positional argument: 'capsys'
    # tests.test_function_2(capsys)        # <--- NameError: name 'capsys' is not defined 
    # tests.test_function_2(pytest.capsys) # <--- AttributeError: module 'pytest' has no attribute 'capsys'
    pass

test_callee_with_instrumentation()
test_callee_with_capsys()

Tenho quase certeza de que os conftest.pyjogos são irrelevantes, mas para completar:

import pytest

@pytest.fixture(scope='function')
def instrumentation():
    return { 'a': 800, 'b': 620, 'c': 44 }

1 No meu código real, o capsysé um dos muitos parâmetros.

2 Uma pergunta semelhante existe aqui . Não é uma duplicata IMHO, porque estou perguntando sobre testes de execução programática , e não sobre o significado apropriado de capsys.

python

2 respostas

  1. Best Answer

    capsysé um acessório do PyTest.

    Você pode listar todos os equipamentos usando:

    pytest --fixtures

    O código-fonte pode ser encontrado em _pytest/capture.py. (Observação: você não deve tentar instanciar/chamar o dispositivo diretamente .)

    Escreva sua função com um argumento:

    import tests

def test_callee_with_instrumentation():
    tests.test_function_1({'a': 110, 'b': 55, 'g': 6000})

def test_callee_with_capsys(capsys):
    tests.test_function_2(capsys)

    Em seguida, execute pytest your_test_file.pye o PyTest instanciará automaticamente o dispositivo e o passará para o teste.

    Se você não quiser usar o PyTest para executar os testes, não tente capsyscapturar a saída padrão.

    from contextlib import redirect_stdout
from io import StringIO

def test_function_1(instrumentation) -> None:
    assert instrumentation['a'] > instrumentation['b']

def test_function_2() -> None:
    with redirect_stdout(StringIO()) as f:
        print("Hello, pytest!")
    captured = f.getvalue()
    assert captured == "Hello, pytest!\n"

    Se você quiser usar o PyTest, mas não quiser chamá-lo diretamente do console, chame-o pytest.main()de dentro do seu código:

    import tests

def test_callee_with_instrumentation():
    tests.test_function_1({'a': 110, 'b': 55, 'g': 6000})

def test_callee_with_output(capsys):
    tests.test_function_2(capsys)

if __name__ == "__main__":
    import pytest
    pytest.main()
    • 1

  2. Se alguém quiser executar seletivamente apenas alguns testes, e se o arquivo subjacente tiver um nome arbitrário, os seguintes ajustes nos pytest.mainargumentos podem ser úteis:

    import tests
import pytest

def test_callee_with_instrumentation():
    tests.test_function_1({'a': 110, 'b': 55, 'g': 6000})

def test_callee_with_capsys(capsys):
    tests.test_function_2(capsys)

if __name__ == "__main__":
    import pytest
    pytest.main([
        '-k',
        'callee_with_capsys', # <--- selectively run just a single test
        '--override-ini',
        f'python_files={__file__}' # <--- collect just this file !
    ])

    Então, quando você executa o script com python(não pytest):

    (BBB) orenishshalom BBB % python main.py 
=============================================================== test session starts ================================================================
platform darwin -- Python 3.13.0, pytest-8.3.3, pluggy-1.5.0
rootdir: /Users/orenishshalom/Downloads/BBB
collected 2 items / 1 deselected / 1 selected                                                                                                      

main.py .                                                                                                                                    [100%]

========================================================= 1 passed, 1 deselected in 0.01s ==========================================================

    1 Vindo de um contexto databricks, ainda tenho dificuldade, pois __file__não é definido ao executar uma célula... arrggghh...

    • 0

relate perguntas