1 import unittest, HTMLTestRunner
2 from parameterized import parameterized # 参数化所需要的模块
3
4 def login(username, passwd): # 需要测试的方法
5 if username == 'nidaye' and passwd == '123456':
6 return True
7 return False
8
9
10 class testLogin(unittest.TestCase):
11 @parameterized.expand( # login 函数的入参参数及预期结果
12 [
13 ('nidaye', '123456', True),
14 ('nidaye', '', False),
15 ('', '', False),
16 ('nidama', '12345', False)
17 ]
18 )
19 def test_Login(self, username, passwd, hope): # 测试用例, unittest会自动循环执行上面多行参数
20 '''这个是登录'''
21 statu = login(username, passwd)
22 self.assertEqual(statu, hope) # 断言用例执行结果
23
24 if __name__ == '__main__':
25 suite = unittest.TestSuite()
26 suite.addTest(unittest.makeSuite(testLogin))
27 fw = open('report.html', 'wb')
28 runner = HTMLTestRunner.HTMLTestRunner(stream=fw, title='nidayetest')
29 runner.run(suite)