python mock

作者：昔日暖阳发布日期:2025-08-22 浏览:86

from unittest import mock

# 示例代码：使用 mock 模拟一个函数的返回值

def get_data():
    return fetch_data_from_api()

@mock.patch('__main__.fetch_data_from_api')
def test_get_data(mock_fetch_data):
    # 设置模拟函数的返回值
    mock_fetch_data.return_value = {'key': 'value'}

    # 调用被测试的函数
    result = get_data()

    # 断言结果是否符合预期
    assert result == {'key': 'value'}
    print("测试通过")

# 解释说明：
# 1. `mock.patch` 是 unittest.mock 模块中的一个装饰器，用于临时替换目标对象（这里是 `fetch_data_from_api` 函数）。
# 2. `mock_fetch_data.return_value` 用于设置模拟函数的返回值。
# 3. `assert` 语句用于验证 `get_data()` 的返回值是否与预期一致。
# 4. 这种方式可以避免在测试中调用真实的 API，从而提高测试的速度和可靠性。

# 注意：实际使用时，需要确保 `fetch_data_from_api` 函数的路径正确。

上一篇：python key value

下一篇：python 字符串比较大小