unittest运行测试的三种方式:

  • unittest.main()
  • TestSuit/TextTestRunner
  • discover()/ TextTestRunner

unittest.main()会执行所有用例

TestSuit/TextTestRunner

灵活测试某些用例

  1. import unittest
  4. class MyTest(unittest.TestCase):
  6.     def test_case(self):
  7.         print("test 2 + 2 ")
  8.         self.assertEqual(2 + 2, 4)
  11. class MyTest2(unittest.TestCase):
  13.     def test_case(self):
  14.         print("test 2 - 2 ")
  15.         self.assertEqual(2 - 2, 0)
  18. if __name__ == '__main__':
  19.     # 测试套件
  20.     suit = unittest.TestSuite()
  21.     suit.addTest(MyTest("test_case"))
  22.     suit.addTest(MyTest2("test_case"))
  24.     # 运行测试套件
  25.     runner = unittest.TextTestRunner()
  26.     runner.run(suit)

discover()/ TextTestRunner

  1. import unittest
  4. suit = unittest.defaultTestLoader.discover("./", "unittest_*.py")  # 执行所有unittest_开头的文件
  6. # 运行测试套件
  7. runner = unittest.TextTestRunner()
  8. runner.run(suit)
  1. # unittest_assert.py
  2. import unittest
  5. class MyClass(unittest.TestCase):
  7.     def test_equal(self):
  8.         user = "admin"
  9.         self.assertEqual(user, "admin")
  10.         self.assertNotEqual("a", "b")
  12.     def test_in(self):
  13.         info = "恭喜，你的邮件已经发送成功！"
  14.         self.assertIn("成功", info)
  15.         self.assertNotIn("失败", info)
  17.     def test_true(self):
  18.         a = False
  19.         b = True
  20.         self.assertTrue(b)
  21.         self.assertFalse(a)
  24. if __name__ == '__main__':
  25.     unittest.main()
  1. # unittest_demo.py
  2. import unittest
  5. class MyClass(unittest.TestCase):
  7.     def test_equal(self):
  8.         user = "admin"
  9.         self.assertEqual(user, "admin")
  10.         self.assertNotEqual("a", "b")
  12.     def test_in(self):
  13.         info = "恭喜，你的邮件已经发送成功！"
  14.         self.assertIn("成功", info)
  15.         self.assertNotIn("失败", info)
  17.     def test_true(self):
  18.         a = False
  19.         b = True
  20.         self.assertTrue(b)
  21.         self.assertFalse(a)
  24. if __name__ == '__main__':
  25.     unittest.main()
  1. # unittest_fixture.py
  2. import unittest
  5. def setUpModule():
  6.     print(">>>>>>>>>>>>>> start module")
  9. def tearDownModule():
  10.     print(">>>>>>>>>>>>>> start module")
  13. class MyClass(unittest.TestCase):
  15.     @classmethod
  16.     def setUpClass(cls):
  17.         print("====> start class")
  19.     @classmethod
  20.     def tearDownClass(cls):
  21.         print("===> end class")
  23.     def setUp(self):
  24.         print("start test")
  26.     def tearDown(self):
  27.         print("end test")
  29.     def test_case(self):
  30.         print("---->test case")
  31.         self.assertEqual(2+2, 4)
  33.     def test_case2(self):
  34.         print("---->test case2")
  35.         self.assertEqual(3, 4)
  38. class MyClass2(unittest.TestCase):
  40.     def test_case3(self):
  41.         print("---> test case3")
  42.         self.assertEqual(3, 3)
  45. if __name__ == '__main__':
  46.     unittest.main()
  1. # unittest_runner.py
  2. import unittest
  5. class MyTest(unittest.TestCase):
  7.     def test_case(self):
  8.         print("test 2 + 2 ")
  9.         self.assertEqual(2 + 2, 4)
  12. class MyTest2(unittest.TestCase):
  14.     def test_case(self):
  15.         print("test 2 - 2 ")
  16.         self.assertEqual(2 - 2, 1)
  19. if __name__ == '__main__':
  20.     # unittest.main()
  21.     # 测试套件
  22.     suit = unittest.TestSuite()
  23.     # suit.addTest(MyTest2("test_case"))
  24.     suit.addTest(MyTest("test_case"))
  26.     # 运行测试套件
  27.     runner = unittest.TextTestRunner()
  28.     runner.run(suit)