代码示例： pytest的简单用法

demo文件目录结构：

 

demo_project/

|----------- __init__.py

|----------- demo_test.py

|----------- test_dbversion.py

|----------- serialize.py

|----------- readme.txt

 

说明：

执行结果中包含print输出。需要注意的是，在pytest中，print输出不能直接使用print("xxx")的方式，因为会被pytest模块归入capsys中。

因为对于pytest而言，print属于sys.stdout，即输出结果的类型。类似的，assert属于sys.stderr的类型

具体的捕获方式参考： https://docs.pytest.org/en/latest/capture.html#capturing-of-the-stdout-stderr-output

简单贴一下官方示例代码：

def test_myoutput(capsys):  # or use "capfd" for fd-level
    print("hello")
    sys.stderr.write("world\n")
    captured = capsys.readouterr()
    assert captured.out == "hello\n"
    assert captured.err == "world\n"
    print("next")
    captured = capsys.readouterr()
    assert captured.out == "next\n"

因此，在我的示例代码中，使用with通过上下文管理，临时关闭capsys，使得print的结果能被打印出来，具体效果参见最后的执行结果示例

 

各文件代码如下：

__init__.py

 #

 

demo_test.py

import pytest


@pytest.mark.test
def test_demo(capsys):
    with capsys.disabled():
        print(host_info_call_data())
    assert host_info_call_data() == '''hostname\tplatform\tcpus    cores\tsockets\tmemory(GB)
rac1    Linux x86 64-bit    2    2    2    7.69    '''
    return 1


def host_info_call_data():
    data = '''hostname    platform    cpus    cores    sockets    memory(GB)
rac1    Linux x86 64-bit    2    2    2    7.69    '''
    return data

 

 

test_dbversion.py

import pytest

from .serialize import serialize


def script_data(num):
    data = {
        1: '''rdbms_version\tcluster_version
11.2.0.4.0\t11.2.0.4.0
''',
        2: '''rdbms_version\tcluster_version
11.2.0.4.0    -
'''
    }
    return data[num]


@pytest.mark.test
@pytest.mark.shells
def test_all_version_exist(capsys):
    data = script_data(1)
    row, column, data = serialize(data)
    with capsys.disabled():
        print(data)
    assert data == [['rdbms_version', 'cluster_version'],
                    ['11.2.0.4.0', '11.2.0.4.0']]

 

 

serialize.py

def serialize(data):
    if isinstance(data, list):
        data = str(data)
    data = data.split('\n')
    if data[-1] == '':
        data = data[:-1]
    data = [i.split('\t') for i in data if isinstance(i, str)]
    row = len(data)
    column = len(data[0])
    return row, column, data

 

 

 

执行结果示例：