第11章
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 get_formatted_city(city, country):
"""Generate a neatly formatted full name."""
full_name = city + ', ' + country
return full_name.title()
test_cities.py
import unittest
from city_functions import get_formatted_city
class NamesTestCase(unittest.TestCase):
def test_formatted_city_name(self):
formatted_name = get_formatted_city('santiago', 'chile')
self.assertEqual(formatted_name, 'Santiago, Chile')
unittest.main()
运行结果
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
[Finished in 0.2s]
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() 通过了。
city_functions.py
def get_formatted_city(city, country, population=''):
"""Generate a neatly formatted full name."""
if population:
full_name = city + ', ' + country + ' - population ' + str(population)
else:
full_name = city + ', ' + country
return full_name.title()
test_cities.py
import unittest
from city_functions import get_formatted_city
class NamesTestCase(unittest.TestCase):
def test_formatted_city_name_1(self):
formatted_name = get_formatted_city('santiago', 'chile')
self.assertEqual(formatted_name, 'Santiago, Chile')
def test_formatted_city_name_2(self):
formatted_name = get_formatted_city('santiago', 'chile', 5000000)
self.assertEqual(formatted_name, 'Santiago, Chile - Population 5000000')
unittest.main()
运行结果
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
[Finished in 0.2s]