Python 单元测试

编写单元测试用例，需要引入unittest 模块

编写的测试类需要从 unittest.TestCase继承

以test开头的方法就是测试方法，不以test开头的方法不是测试方法，运行时不被执行

对每一类测试都需要编写test_xxx()方法，调用unittest.TestCase的内置条件判断方法来判断结果，最常用的是assertEqual()

self.assertEqual(abs(-1), 1) #前后相等，测试结果为ok

另一个重要的断言就是期待抛出指定类型的error：

with self.assertRaises(typeError)

运行单元测试，最简单的方法是在最后加上两行代码：

if __name__ == '__main__':

unittest.main()

可以在单元测试编写两个特殊的方法，setUp()和tearDown()，这两个方法分别会在每调用一个测试方法的前后会执行

 

def setUp(self):

　　pass

def tearDown(self):

　　pass

 

实例：

class Student(object):
    def __init__(self, name, score):
        self.name = name
        self.score = score

    def get_grade(self):
        if not isinstance(self.score, int):
            raise ValueError
        if self.score > 100 or self.score < 0:
            raise ValueError
        if self.score >= 80:
            return 'A'
        if self.score >= 60:
            return 'B'
        return 'C'

# _*_ coding:UTF-8 _*_
import unittest
from test2 import Student


class TestStudent(unittest.TestCase):
    def setUp(self):
        print('setup')

    def tearDown(self):
        print('teardown')

    def test_80_100(self):
        s1 = Student('Bat', 80)
        s2 = Student('Sandy', 100)
        self.assertEqual(s1.get_grade(), 'A')
        self.assertEqual(s2.get_grade(), 'A')

    def test_60_80(self):
        s3 = Student('Boo', 79)
        s4 = Student('Carry', 60)
        self.assertEqual(s3.get_grade(), 'B')
        self.assertEqual(s4.get_grade(), 'B')

    def test_0_60(self):
        s5 = Student('xiaoming', 59)
        s5 = Student('xiaolan', 0)
        self.assertEqual(s5.get_grade(), 'C')
        self.assertEqual(s5.get_grade(), 'C')

    def test_invalid(self):
        s5 = Student('xiaoming', -1)
        s6 = Student('xiaolan', 101)
        with self.assertRaises(ValueError):
            s5.get_grade()
        with self.assertRaises(ValueError):
            s6.get_grade()

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