测试是用来确定代码在面对各种输入时能按照要求进行工作，提前测试有利于防止新代码不会破坏程序既有的行为，本章使用unittest中的工具，编写测试用例，核实输入将得到的输出

formatted 格式化 Generate生成 neatly 整洁的 full完全的

 1 from name_function import get_formatted_name #从name_function.py中导入get_formatted_name() 

11-1城市和国家：编写一个函数，它接受两个形参：一个城市名和一个国家名。这个函数返回一个格式为City, Country的字符串，如Santiago, Chile。将这个函数存储在一个名为city _functions.py的模块中。创建一个名为test_cities.py的程序，对刚编写的函数进行测试（别忘了，你需要导入模块unittest以及要测试的函数）。编写一个名为test_city_country()的方法，核实使用类似于'santiago'和'chile'这样的值来调用前述函数时，得到的字符串是正确的。运行test_cities.py，确认测试test_city_country()通过了。

 1 #city_functions.py
 2 def city_country(city_name, country_name, middle=','):  #
 3     """输出城市名和国家名,例如Santiago, chile"""
 4     full_city_country = city_name + middle + country_name
 5     return full_city_country.title()
 6 
 7 #city_name_country_name.py
 8 from city_functions import city_country
 9 print("Enter 'q' at any time to quit.")
10 while True:
11     city_name = input("亲输入城市名: ")
12     if city_name == 'q':
13         break
14     country_name = input("亲输入国家名: ")
15     if country_name == 'q':
16         break
17 
18     formatted_name = city_country(city_name, country_name)
19     print("\n城市和国家名的输出结果: " + formatted_name)
20 
21 
22 #test_cityname_and_country_name.py
23 import unittest                     #导入inttest模块中的工具
24 from city_functions import city_country    
25 
26 class NamesTestCase(unittest.TestCase):    #这个命名可以随意，但通常都与测试的函数相关而且要包含Test字样
27     """测试city_functions.py"""            #继承TestCase类这样Python才知道如何运行你编写的测试，可以认为是一个接口将你的函数
28                                    #在另一个地方运行
29     def test_city_country_name(self):     #这个测试文件的方法要用test_为开头，_test为开头的方法在文件运行时会自动运行
30         """能够正确地处理像Santigo, Chile这类的名字吗?"""
31         formatted_name = city_country('Santigo', 'Chile') #调用了要测试的函数，并将返回值储存到formatted_name中
32         self.assertEqual(formatted_name, 'Santigo,Chile')#使用断言方法assertEqual用来核实得到的结果是否与期望的结果一致
33                         
34   '''在这里，我们知道get_formatted_name()应返回这样的城市国家名，即
35 首字母为大写，且它们之间有一个‘，’，因此我们期望formatted_name的值为Santigo，Chile。为检查是否确实如此，我们调用unittest的方法assertEqual()，并向它传递formatted_
36 name和'Santigo Chile'。代码行self.assertEqual(formatted_name, 'Janis Joplin')的意思是说：“将formatted_name的值同字符串'Santigo,Chile'进行比较，如果它们相等，就万事大吉，如果它
37 们不相等，跟我说一声！'''
38 unittest.main()

11-2人口数量：修改前面的函数，使其包含第三个必不可少的形参population，并返回一个格式为City, Country – population xxx的字符串，如Santiago, Chile –population 5000000。运行test_cities.py，确认测试test_city_country()未通过。修改上述函数，将形参population设置为可选的。再次运行test_cities.py，确认测试test_city_country()又通过了。再编写一个名为test_city_country_population()的测试，核实可以使用类似于'santiago'、'chile'和'population=5000000'这 样 的 值 来 调 用 这 个 函 数 。 再 次 运 行test_cities.py，确认测试test_city_country_population()通过了。

 1 #写到一般才看到是测试两个。。一个可选，一个不可选
 2 #不可选
 3 def city_country_population(city_name, country_name, population='population-50000'):
 4     """输出城市名和国家名,例如Santiago, chile"""
 5     full_city_country = (city_name + ', ' +
 6                          country_name + str(population))
 7     return full_city_country.title()
 8 #city_country_population_name.py
 9 from city_country_population import city_country_population
10 population = 'population-500000'
11 print("Enter 'q' at any time to quit.")
12 while True:
13     city_name = input("亲输入城市名: ")
14     if city_name == 'q':
15         break
16     country_name = input("亲输入国家名: ")
17     if country_name == 'q':
18         break
19     #population = input("请输入人口数: ")
20     #if population == 'q':
21         #break
22 
23     formatted_name = city_country_population(city_name, country_name, population)
24     print("\n城市和国家名和人口的输出结果: " + formatted_name)
25 #可选
26 #city_country_population.py
27 def city_country_population(city_name, country_name, population):
28     """输出城市名和国家名,例如Santiago, chile"""
29     full_city_country = (city_name + ', ' +
30                          country_name + ' -population ' + population)
31     return full_city_country.title()
32 
33 #city_country_population_name.py
34 from city_country_population import city_country_population
35 print("Enter 'q' at any time to quit.")
36 while True:
37     city_name = input("亲输入城市名: ")
38     if city_name == 'q':
39         break
40     country_name = input("亲输入国家名: ")
41     if country_name == 'q':
42         break
43     population = input("请输入人口数: ")
44     if population == 'q':
45         break
46 
47     formatted_name = city_country_population(city_name, country_name, population)
48     print("\n城市和国家名和人口的输出结果: " + formatted_name)
49 
50 #test_city_and_country_populations.py
51 import unittest
52 from city_country_population import city_country_population
53 
54 class NamesTestCase(unittest.TestCase):
55     """测试city_functions.py"""
56 
57     def test_city_country_population_name(self):
58         """能够正确地处理像Santigo, Chile这类的名字吗?"""
59         formatted_name = city_country_population('Asd', 'Asd', '456')
60         self.assertEqual(formatted_name, 'Asd, Asd -Population 456')
61 
62 unittest.main()
63 #测试不通过的没写，因为太多不通过的方法了，没有明确定义用啥

断言方法

assertEqual(a, b)        核实a == b
assertNotEqual(a, b)     核实a != b
assertTrue(x)            核实x为True
assertFalse(x)           核实x为False
assertIn(item, list)     核实item在list中
assertNotIn(item, list)  核实item不在list中

11-3雇员：编写一个名为Employee的类，其方法__init__()接受名、姓和年薪，并将它们都存储在属性中。编写一个名为give_raise()的方法，它默认将年薪增加5000美元，但也能够接受其他的年薪增加量。为Employee编写一个测试用例，其中包含两个测试方法：test_give_default_raise()和test_give_custom_raise()。使用方法setUp()，以免在每个测试方法中都创建新的雇员实例。运行这个测试用例，确认两个测试都通过了。

 1 #Employee.py
 2 class Employee:
 3       """收集打工人的信息"""
 4       def __init__(self, firstname, lastname, money):
 5           """打工人初始化"""
 6           self.firstname = firstname
 7           self.lastname = lastname
 8           self.money = money
 9      def give_raise(self, addmoney=5000):
10           """应该加的钱(可能吗)"""
11           self.money += addmoney
12 
13 #测试用例
14 import inttest
15 from Employee import Employee     #文件Employee内的Employee类         
16 class EmployeeTestCase(unittest.TestCase):              #如果你在TestCase类中包含了方法setUp()，Python将先运行
17                                                    # 它，再运行各个以test_打头的方法
18       """测试不同打工人的信息"""
19       def setUp(self):
20            """初始化打工人信息"""                        #方法setUp()，让我们只需创建这些对象一
21                                                         # 次，并在每个测试方法中使用它们。
22            self.default = Employee('xiao', 'ma', 1000) #变量名包含前缀self（即存储在属性中）
23           """默认加的薪水"""
24           self.default.give_raise()
25           self.assertEqual(self.default.money,6000)
26       def test_igive_coustom_raise(self):
27           """特殊加薪"""
28           self.default.give_raise(8000)
29           self.assertEqual(self.default.money,9000)
30 unittest.main()

小结：测试代码

参与工作量较大的项目时，你应对自己编写的函数和类的重要行为进行测试，在项目早期，不要试图去编写全覆盖的测试用例，除非有充分的理由这样做。

使用模块unittest中的工具来为函数和类编写测试；如何编写继承unittest.TestCase的类，以及如何编写测试方法，以核实函数和类的行为符合预期；如何使用方法setUp()来根据类高效地创建实例并设置其属性，以便在类的所有测试方法中都可使用它们。