unittest是否适用于Python中的功能（非面向对象）程序？

Question

我是一个初学者程序员，在过去的一周里，我一直在尝试编写单元测试。我通读了单元测试文档，并观看了两个关于使用Mock实现单元测试的长教程。文档广泛地引用了模拟类，但对于函数，我不确定是否应该使用@patch/with patch、patch.dict{}、side_effect或其他选项来模拟函数，特别是函数的参数

mymodule.py

def regex():
    '''Runs a regex, creates a dict 'data' and then calls scraper(data)'''

def scraper(data):
    '''scrapes a website and then calls a function which submits data to a db'''

我想创建一个测试，将测试数据传递给函数scraper。先谢谢你

Answer 1

是的，您还可以使用mock对非面向对象代码进行单元测试

见下例：

from unittest.mock import MagicMock
def a():
    return 10
def b():
    print(a())  
b()
a = MagicMock(return_value=3)
b()

输出为：

10
3

在上一个示例中，mock用于伪造/模拟函数a（），因此您可以单独测试函数b（），b（）是SUT，而无需调用（）实际实现。这对于更复杂的代码非常有用，特别是当函数a（）依赖于单元测试级别可能不可用的数据时