Mock 使用

from unittest.mock import Mock, patch

def test_api_call():
   with patch('requests.get') as mock_get:
       mock_get.return_value.json.return_value = {'status': 'ok'}
       # 测试代码
       result = api_call()
       assert result['status'] == 'ok'

参数化测试

import pytest

@pytest.mark.parametrize("input,expected", [
   (2, 4),
   (3, 9),
   (4, 16),
])
def test_square(input, expected):
   assert input * input == expected

转载请注明：周志洋的博客 » Python进阶-测试进阶