python测试断言

　　这几天看了一下python的测试，基于函数方法和基于类的测试，主要使用的模块是unittest模块，为特定的方法和类，建立测试模块，测试函数功能是否满足预期。下面是模仿书里面的一个例子。

　　City.py模块

def city(city_name,country_name):
    full_city_name = city_name +','+country_name
    return full_city_name

　　为这个模块创建一个测试模块，主要是为了测试这个city方法是否能够满足预期。

　　assert模块

import unittest
from City import city
class Testcity(unittest.TestCase):

    def test_city(self):
        city_name = 'Santiago'
        country_name = 'Chile'
        full_city_name = city(city_name,country_name)
        print(full_city_name)
        self.assertEqual(full_city_name,'Santiago,Chile')
        # self.assertEqual(4+5,9)

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

　　主要就是导入unittest模块，然后定义一个测试类，继承unittest.TestCase，在测试类中定义测试方法，针对于被测试方法，编写测试用例，然后assert断言，查看输入，如果在运行过程中出现报错，则证明模块功能有问题，需要修改功能。如果测试没有问题，则证明测试能够通过，功能模块没有问题。

　　

　　这就是测试成功的结果，如果测试部成果，如下图所示。

　　

　　会抛出一个异常，在断言的地方。然后可以根据这个异常，来进行排错。

　　

　　一般编写的代码，都需要经过测试之后，才能正常使用。避免复杂的人工测试，assert断言能够做到自动化，一旦有报错，则直接能判断功能是否正常。

　　下面是一个对类的测试断言，写在同一个模块中，只是为了说明测试方法。

class Employee():
    def __init__(self,first_name,last_name,salary):
        self.first_name = first_name
        self.last_name = last_name
        self.salary = int(salary)

    def give_raise(self,raise_default = 5000):
        self.salary += raise_default

class TestClassEmployee(unittest.TestCase):
    def setUp(self):
        self.username = Employee('John','Smith','')

    def test_give_default_raise(self):
        self.username.give_raise()
        new_raise = 65000
        self.assertEqual(new_raise,self.username.salary)

    def test_give_custom_raise(self):
        self.username.give_raise(7000)
        new_raise = 67000
        self.assertEqual(new_raise, self.username.salary)
if __name__ == '__main__':
    unittest.main()

　　定义了一个员工类，需要知道员工的名字和薪水，同时定义了一个涨工资方法，默认值是5000，在定义测试类的时候，首先定义了个测试对象，通过setUp方法，生产测试对象，后续的测试方法中，可以通过该测试对象来做测试。定义了两个测试方法，默认值和自定义值。测试结果如下：