selenium - unittest一个简单的demo

calculate.py文件
1 class Count:
2     def __init__(self, a, b):
3         self.a = a
4         self.b = b
5 
6     def add(self):
7         return self.a + self.b

 test.py文件
 1 from calculate import Count
 2 import unittest           # 引入unittest模块
 3 
 4 class TestAdd(unittest.TestCase):   # 创建TestAdd类，继承TestCase类
 5     def setUp(self):       # setUp()方法，用于测试用例执行前的初始化工作
 6         print(‘start‘)
 7 
 8     def tearDown(self):    # tearDown()方法，用于测试用例执行之后的善后工作
 9         print(‘end‘)
10 
11     def test_add_1(self):
12         print(‘add 1‘)
13         self.assertEqual(Count(2, 4).add(), 6)  # 使用unittest框架提供的assertEqual()方法进行断言，assertEqual()方法由TestCase类继承而来
14 
15 
16 if __name__ == ‘__main__‘:
17     unittest.main()