Python | 如何将测试输出到标准输出

在编写Python测试时，我们常常需要将某些数据或信息输出到标准输出中，方便调试或查看测试结果。下面就来介绍几种Python中实现测试输出到标准输出的方法。

print函数

最简单、最直接的方法就是使用Python内置的print函数。我们可以通过在需要输出的地方调用print函数，将信息打印到标准输出中。例如：

def test_add():
    a = 1
    b = 2
    c = add(a, b)
    print("a={}, b={}, a+b={}".format(a, b, c))
    assert c == 3

在这个例子中，我们在add函数的测试中使用了print函数，将测试信息输出到了标准输出中。

logging模块

print函数虽然简单，但是并不足够强大，特别是在需要输出的信息比较复杂、需要按照一定格式记录的情况下。这时候就需要使用logging模块来实现测试输出到标准输出。

logging是Python内置的日志模块，可以实现打印输出、文件输出等多种输出方式，并支持按照不同级别进行分类输出。在进行测试输出时，我们可以使用logging模块，在需要输出的地方调用logging函数，将信息打印到标准输出中。

例如，我们可以定义一个Logger实例，并在测试中使用它来输出信息：

import logging
 
def test_add():
    a = 1
    b = 2
    c = add(a, b)
    logger = logging.getLogger(__name__)
    logger.info("a={}, b={}, a+b={}".format(a, b, c))
    assert c == 3

在这个例子中，我们定义了一个Logger实例，并使用它来记录测试信息。其中，用到了Logger的info级别，表示记录信息。

unittest模块

unittest模块是Python标准库中的测试框架，它提供了一系列工具和接口，方便编写和运行测试用例。在进行测试输出时，我们也可以利用unittest模块提供的工具，将测试信息输出到标准输出中。

例如，我们可以使用unittest.TestCase的assert方法，在测试不通过时输出信息：

import unittest
 
class TestAdd(unittest.TestCase):
    def test_add(self):
        a = 1
        b = 2
        c = add(a, b)
        self.assertEqual(c, 3, "a={}, b={}, a+b={}".format(a, b, c))

在这个例子中，我们通过assertEqual方法来测试add函数的正确性，并在测试不通过时调用其第三个参数，将信息输出到标准输出中。

以上就是Python中实现测试输出到标准输出的几种方法。在实际编写测试时，我们可以根据需要选择适合自己的方法，方便地输出测试信息。