使用python编写单元测试

上一篇 / 下一篇  2022-07-04 10:39:07 / 个人分类:python

使用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()
#运行单元测试
if __name__ == '__main__':
unittest.main()

3、运行结果如下:

1.png

4、行单元测试另一种方法:

在命令行通过参数-m unittest直接运行单元测试,例如:python -m unittest student_test

2.png

最后对使用unittest模块的一些总结:

  1. 编写单元测试时,需要编写一个测试类,从unittest.TestCase继承

  2. 对每一个类测试都需要编写一个test_xxx()方法

  3. 最常用的断言就是assertEqual()

  4. 另一种重要的断言就是期待抛出指定类型的Error,eg:with self.assertRaises(KeyError):

  5. 另一种方法是在命令行通过参数-m unittest直接运行单元测试:eg:python -m unittest student_test

  6. 最简单的运行方式是xx.py的最后加上两行代码:

if __name__ == '__main__':
    unittest.main()

TAG:

 

评分:0

我来说两句

日历

« 2022-08-08  
 123456
78910111213
14151617181920
21222324252627
28293031   

数据统计

  • 访问量: 37984
  • 日志数: 263
  • 建立时间: 2021-06-21
  • 更新时间: 2022-08-08

RSS订阅

Open Toolbar