使用python3.6怎么实现一个单元测试功能

本篇文章为大家展示了使用python3.6怎么实现一个单元测试功能，内容简明扼要并且容易理解，绝对能使你眼前一亮，通过这篇文章的详细介绍希望你能有所收获。

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、运行结果如下：

4、行单元测试另一种方法：在命令行通过参数-m unittest直接运行单元测试，例如：python -m unittest student_test

最后对使用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()

上述内容就是使用python3.6怎么实现一个单元测试功能，你们学到知识或技能了吗？如果还想学到更多技能或者丰富自己的知识储备，欢迎关注亿速云行业资讯频道。