Python自动化测试：unittest与pytest框架

在Python中，unittest和pytest都是常用的自动化测试框架。它们提供了编写测试用例、测试套件和执行测试的强大功能。

1. unittest框架

unittest是Python标准库的一部分，因此无需额外安装。它提供了丰富的断言方法，用于验证测试结果。

示例代码：

python复制代码

	import unittest
	class TestStringMethods(unittest.TestCase):
	def test_upper(self):
	self.assertEqual('foo'.upper(), 'FOO')
	def test_isalpha(self):
	self.assertTrue('foo'.isalpha())
	self.assertFalse('foo123'.isalpha())
	def test_split(self):
	s = 'hello world'
	self.assertEqual(s.split(), ['hello', 'world'])
	# 使用断言检查列表长度
	with self.assertRaises(ValueError):
	s.split(maxsplit=1)
	if __name__ == '__main__':
	unittest.main()

在这个示例中，我们定义了一个名为TestStringMethods的测试类，其中包含三个测试方法。每个测试方法都以test_开头，这是unittest的一个约定。assertEqual和assertTrue是断言方法，用于验证预期结果与实际结果是否一致。

2. pytest框架

pytest是一个更简洁、更易于使用的测试框架。它不需要继承任何基类或编写特定的测试方法。

示例代码：

首先，确保你已经安装了pytest：

bash复制代码

pip install pytest

然后，创建一个名为test_example.py的测试文件，并编写以下代码：

python复制代码

	def add(x, y):
	return x + y
	def test_add():
	assert add(1, 2) == 3
	assert add(0, 0) == 0
	with pytest.raises(TypeError):
	add(1, '2')
	def test_subtract():
	assert add(5, -3) == 2

在这个示例中，我们定义了一个简单的add函数，然后创建了两个测试函数test_add和test_subtract。每个测试函数都以test_开头，这是pytest的一个约定。assert语句用于验证预期结果与实际结果是否一致。如果assert语句失败，测试将被视为失败。

要运行这些测试，请在命令行中导航到包含测试文件的目录，并执行以下命令：

bash复制代码

pytest

pytest将自动查找并执行所有以test_开头的函数。如果所有测试都通过，则不会显示任何输出。如果有测试失败，pytest`将显示失败的详细信息。