主页 / coding / 问题 / 79181238
Accepted
OrenIshShalom
Asked: 2024-11-12 21:08:26 +0800 CST

使用 capsys 选择性调用测试函数

  • 772

我可以选择性地运行test_function_1，覆盖我的conftest.py装置中的仪表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"

当我尝试调用时test_function_2，我不知道如何传递capsys给它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()

我很确定这些conftest.py装置是无关紧要的，但为了完整起见：

import pytest

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

1在我的实际代码中，是众多capsys参数之一。

2这里有类似的问题。在我看来，这不是重复的，因为我问的是编程运行测试，而不是的正确含义capsys。

python

2 个回答

  1. Best Answer

    capsys是一个 PyTest 装置。

    您可以使用以下方式列出所有装置：

    pytest --fixtures

    源代码位于_pytest/capture.py。（注意：您不应该尝试直接实例化/调用装置。）

    编写带有参数的函数：

    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)

    然后运行pytest your_test_file.py，PyTest 将自动实例化该装置并将其传递到测试中。

    如果您不想使用 PyTest 运行测试，那么就不要capsys尝试捕获标准输出。

    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"

    如果您确实想使用 PyTest，但不想直接从控制台调用 PyTest，那么请pytest.main()从代码中调用：

    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. 如果想要有选择地只运行某些测试，并且底层文件具有任意名称，则以下pytest.main参数调整可能会有所帮助：

    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 !
    ])

    python然后当您使用（而不是）运行脚本时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来自 databricks 上下文，我仍然很难，因为__file__在运行单元格时没有定义......arrggghh......

    • 0

相关问题