本文章整理自：https://blog.csdn.net/dou_being/article/details/80039609 使用python3.6编写一个单元测试demo，例如：对学生Student类编写一个简单的单元测试。 1、编写Student类：
#!/usr/bin/env python3 # -*- coding: utf-8 -*- class Student(object): def __init__(self,name,score): self.name = name self.score = score def get_grade(self): if self.score >= 80 and self.score <= 100: return 'A' elif self.score >= 60 and self.score <= 79: return 'B' elif self.score >= 0 and self.score <= 59: return 'C' else: raise ValueError('value is not between 0 and 100')2、编写一个测试类TestStudent，从unittest.TestCase继承：
#!/usr/bin/env python3 # -*- coding: utf-8 -*- import unittest from student import Student class TestStudent(unittest.TestCase): def test_80_to_100(self): s1 = Student('Bart',80) s2 = Student('Lisa',100) self.assertEqual(s1.get_grade(),'A') self.assertEqual(s2.get_grade(),'A') def test_60_to_80(self): s1 = Student('Bart',60) s2 = Student('Lisa',79) self.assertEqual(s1.get_grade(),'B') self.assertEqual(s2.get_grade(),'B') def test_0_to_60(self): s1 = Student('Bart',0) s2 = Student('Lisa',59) self.assertEqual(s1.get_grade(),'C') self.assertEqual(s2.get_grade(),'C') def test_invalid(self): s1 = Student('Bart',-1) s2 = Student('Lisa',101) with self.assertRaises(ValueError): s1.get_grade() with self.assertRaises(ValueError): s2.get_grade()3、运行单元测试代码文件时，要在命令行中操作。示意如下： 4、行单元测试另一种方法：在命令行通过参数-m unittest直接运行单元测试，例如：python -m unittest student_test 最后对使用unittest模块的一些总结： 编写单元测试时，需要编写一个测试类，从unittest.TestCase继承 对每一个类测试都需要编写一个test_xxx()方法 最常用的断言就是assertEqual() 另一种重要的断言就是期待抛出指定类型的Error，eg：with self.assertRaises(KeyError): 另一种方法是在命令行通过参数-m unittest直接运行单元测试：eg：python -m unittest student_test 最简单的运行方式是xx.py的最后加上两行代码：
if __name__ == '__main__': unittest.main()关于pytest的用法，请参考链接：https://www.cnblogs.com/zongfa/p/9245038.html
