Python单元测试

本文章整理自：https://blog.youkuaiyun.com/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