文章目录
11-1 城市和国家
编写一个函数，它接受两个形参：一个城市名一个国家名。这个函数返回一个格式为City, Country的字符串，如Santiago, Chile。将这个函数存储在一个名为city_functions.py的模块中。
创建一个名为test_cities.py的程序，对刚编写的函数进行测试（别忘了，你需要导入模块unittest以及要测试的函数）。编写一个名为test_city_country()的方法，核实使用类似于‘santiago’和‘chile’这样的值来调用前述函数时，得到的字符串是正确的。运行test_cities.py, 确认测试test_city_country()通过了。
city_functions.py
def place(city, country):
"""生成一个地点"""
place = city.title() + ", " + country.title()
return place
test_city.py
import unittest
from city_functions import place
class PlaceTestCase(unittest.TestCase):
"""测试city_functions.py"""
#继承unittest模块中的TestCase类
def test_city_country_name(self):
"""能够正确处理像Santiago, Chile这样的地名吗？"""
place_name = place('santiago', 'chile')
self.assertEqual(place_name, 'Santiago, Chile')
unittest.main()
build
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
11-2 人口数量
修改前面的函数，使其包含第三个不可少的形参population，并返回一个格式为City, Country - population xxx的字符串，如Santiago, Chile - population 5000000。运行test_city.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）修改函数模块
city_functions2.py
def place(city, country, population = ''):
"""生成一个地点和该地人口"""
if population:
place = city.title() + ", " + country.title() + " - population " + str(population)
else:
place = city.title() + ", " + country.title()
return place
（2）测试
import unittest
from city_functions2 import place
class PlacePopulationTestCase(unittest.TestCase):
"""测试city_functions2.py"""
def test_city_country_name(self):
"""能够正确处理像Santiago, Chile这样的地名吗？"""
place_name = place('santiago', 'chile')
self.assertEqual(place_name, 'Santiago, Chile')
def test_city_country_population(self):
"""能够正确处理像Santiago, Chile - population 5000000这样的地名和信息吗？"""
place_name = place('santiago', 'chile', population = '5000000')
self.assertEqual(place_name, 'Santiago, Chile - population 5000000')
unittest.main()
注意：
- 记住导入整个模块（unittest）的语句。记住从模块导入函数的语句。
- 【导入整个模块的情况下调用函数/类要加模块名的前缀，如unittest.TestCase, unittest.main()】
- 【导入函数的情况下不用加前缀】
- 记住继承的类名。unittest.TestCase。这是固定的。
- self.assertEqual() 断言方法
- unittest.main()
11-3 雇员
编写一个名为Employee的类，其方法__init__（接受名、姓和年薪，并将它们都存储在属性中。编写一个名为give_raise()的方法，它默认将年薪增加5000美元，但也能够接受其他的年薪增量。
为Employee编写一个测试用例，其中包含两个测试方法：test_give_default_raise()和test_give_custom_raise()。使用方法setUp()，以免在每个测试方法中都创建新的雇员实例。运行这个测试用例，确认两个测试都通过了。
（1）编写一个名为Employee的类
class Employee():
"""收集雇员的信息"""
def __init__(self, last_name, first_name, wage):
"""存储雇员的名，姓，薪水"""
self.last_name = last_name.title()
self.first_name = first_name.title()
self.wage = wage
def give_raise(self, increment=5000):
self.increment = increment
self.wage += increment
（2）测试
import unittest
from employee import Employee
class TestEmployee(unittest.TestCase):
"""针对Employee类的测试"""
def setUp(self):
"""创建一组名，姓，薪水的实例，供使用的测试方法使用"""
f_name = "Wang"
l_name = "Xizhi"
wage = 10000
self.wangxizhi = Employee(l_name, f_name, wage)
def test_give_default_raise(self):
"""测试使用默认增量能被妥善存储"""
self.wangxizhi.give_raise()
self.assertEqual(self.wangxizhi.wage, 15000)
def test_give_custom_raise(self):
"""测试使用其他年薪增量能被妥善存储"""
self.wangxizhi.give_raise(2000)
self.assertEqual(self.wangxizhi.wage, 12000)
unittest.main()
build
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
注意：
①方法setUp()
②包含前缀self的变量名，可以在这个类的任何地方使用。