python unittest自动测试框架
编写函数或者类时进行测试,确保代码正常工作
python unittest 模块提供了代码测试工具。按照定义测试包括两部分:管理测试依赖库的代码(称为‘固件’)和测试本身。
单元测试用于核实函数的某个方面没有问题;
测试用例是一组单元测试,这些单元测试一起核实函数在各种情况选的行为都符合要求
# Author:song import unittest class TestString(unittest.TestCase): def test_upper(self): self.assertEqual('Foo'.upper(),'FOO') def test_isupper(self): self.assertTrue('FOO'.isupper()) self.assertFalse('Foo'.isupper()) def test_split(self): s= 'hello world' self.assertEqual(s.split(),['hello','world']) with self.assertRaises(TypeError): s.split(2) if __name__=="__main__": #运行测试 unittest.main()
unittest.main():使用它可以将一个单元测试模块变为可直接运行的测试脚本,main()方法使用TestLoader类来搜索所有包含在该模块中以“test”命名开头的测试方法,并自动执行
断言真值
大部分测试会断言某个条件的真值
Python在unittest.TestCase类中提供了很多断言方法
常见几个
测试固件
固件是测试所需要的外部资源。例如一个类的所有测试可能需要另一个类的实例(用来提供配置设置)或另一个共享资源。配置固件需要覆盖setUp()。完成清理需要覆盖tearDown()
import unittest class FixturesTest(unittest.TestCase): def setUp(self): print('in setup') self.fixture = range(1,10) def tearDown(self): print('in teardown') del self.fixture def test(self): print('in test') self.assertEqual(self.fixture,range(1,10)) if __name__ =="__main__": unittest.main()
结果:
in setup
in test
in teardown
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
unittest.TestCase类包含方法setUp(),让我们只需创建这些对象一 次,并在每个测试方法中使用它们。
AnonymousSurvey.py
class AnonymousSurvey():#创建匿名调查的类 def __init__(self,question): self.question = question self.responses = [] def show_question(self): print(self.question) def store_response(self,new_response): self.responses.append(new_response) def show_results(self): print('Survey result:') for response in self.responses: print('-'+response)
测试代码,
import unittest class TestAnonymousSurvey(unittest.TestCase): def setUp(self): question = "What language did you first learn to speak?" self.my_survey = AnonymousSurvey(question) self.responses = ['English', 'Spanish', 'Mandarin'] def test_store_single_response(self): #测试单个答案会被妥善地存储 self.my_survey.store_response(self.responses[0]) self.assertIn(self.responses[0], self.my_survey.responses) def test_store_three_responses(self): #测试多个答案被妥善存储 for response in self.responses: self.my_survey.store_response(response) for response in self.responses: self.assertIn(response, self.my_survey.responses) unittest.main()
方法setUp()做了两件事情:创建一个调查对象;创建一个答案列表。存储这两样东西的变量名包含前缀self(即存储在属性中),因此可在这个类的任何地方使用。这让两个测试方法都更简单,因为它们都不用创建调查对象和答案。方法setUp()让测试方法编写起来更容易,相比于在每个测试方法中都创 建实例并设置其属性,这要容易得多。
运行结果
.. ---------------------------------------------------------------------- Ran 2 tests in 0.000s OK
运行测试用例时,每完成一个单元测试,Python都打印一个字符:测试通过时打印一个 句点;测试引发错误时打印一个E;测试导致断言失败时打印一个F。这就是你运行测试 用例时,在输出的第一行中看到的句点和字符数量各不相同的原因。如果测试用例包含 很多单元测试,需要运行很长时间,就可通过观察这些结果来获悉有多少个测试通过了
测试结果
ok:测试通过。
FAIL:测试没有通过产生了一个AssertionError异常。
ERROR:测试产生了AssertionError以外的一个异常。