unittest 原名为 PyUnit,是由 java 的 JUnit 衍生而来
unittest库提供了test cases, test suites, test fixtures,test runner。
- test cases 通过继承TestCase类,创建一个test或者多个test
- test suites 测试套件,相当于test的容器
- test fixtures 通常为setup + test case + teardown结构
- test runner 执行测试用例tests或suit,可以指定执行方式
待测试的类:
class Count:
def __init__(self,a,b):
self.a = a
self.b = b
def add(self):
return self.a+self.b
示例1:
import unittest
class TestDemo(unittest.TestCase):
def setUp(self):
print("setUp")
def test_add(self):
z = Count(2,3)
self.assertEqual(z.add(),5)
def test_invid_add(self):
z = Count(2, 3)
self.assertNotEqual(z.add(),6)
def tearDown(self):
print("tearDown")
if __name__=="__main__":
suit=unittest.TestSuite()#在pycharm中,手动修改以一般python文件方式去运行
suit.addTest(TestDemo("test_invid_add"))
suit.addTest(TestDemo("test_add"))
runner=unittest.TextTestRunner()
runner.run(suit)
# unittest.main() #默认按照字母和数字顺序执行,等同于unittest.makeSuite(test)
示例2:
import unittest
lass TestDemo(unittest.TestCase):
@classmethod
def setUpClass(cls):
print("setUpClass")
def test_add(self):
print("test_add")
z = Count(2,3)
self.assertEqual(z.add(),5)
def test_invid_add(self):
print("test_invid_add")
z = Count(2, 3)
self.assertNotEqual(z.add(),6)
@classmethod
def tearDownClass(cls):
print("tearDownClass")
if __name__=="__main__":
suit=unittest.TestSuite()
suit.addTest(TestDemo("test_time"))
suit.addTest(TestDemo("test_add"))
suit.addTest(TestDemo("test_invid_add"))
runner=unittest.TextTestRunner()
runner.run(suit)
从以上2个示例可知道:
- unittest默认是按照字母和数字的顺序运行,倘若需要按照我们指定的顺序执行,可使用
suit.addTest()
的方式去指定 - setUp()、tearDown()会在每个测试用例(test开头)之前和之后执行
- setUpClass(cls)、tearDownClass(cls)仅在用例之前和之后各执行1次,与上述的setUp()、tearDown()不同
- 使用pycharm IDE运行unittest文件,不管是默认的用例执行顺序,还是指定的用例执行顺序,用例执行结果均为默认的顺序,即按照字母和数字的顺序执行。这算是pycharm的一个小小bug,具体做法:进入运行配置界面,在Python 下手动创建指定执行的py文件,而不是Python tests下