Python 编程从入门到实践 11-1 11-2
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
import unittest
from city_function import get_city_conuntry
class NamesTestCase(unittest.TestCase):
"""测试city_function.py"""
def test_city_country(self):
"""能够正确处理像Santiago, Chile这样的名称吗？"""
formatted_name = get_city_conuntry('Santiago', 'Chile')
self.assertEqual(formatted_name, 'Santiago, Chile')
if __name__ == "__main__":
unittest.main()
test_cities.py
def get_city_conuntry(city, country):
"""生成整洁的名称"""
full_name = city + ', ' + country
return full_name.title()
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_city_conuntry(city, country, population=''):
"""生成整洁的名称"""
if population:
full_name = city + ', ' + country + ' - population ' + population
else:
full_name = city + ', ' + country
return full_name
test_cities.py
import unittest
from city_function import get_city_conuntry
class NamesTestCase(unittest.TestCase):
"""测试city_function.py"""
def test_city_country(self):
"""能够正确处理像Santiago, Chile这样的名称吗？"""
formatted_name = get_city_conuntry('Santiago', 'Chile')
self.assertEqual(formatted_name, 'Santiago, Chile')
def test_city_country_population(self):
formatted_name = get_city_conuntry('Santiago', 'Chile', population='5000')
self.assertEqual(formatted_name, 'Santiago, Chile - population 5000')
if __name__ == "__main__":
unittest.main()