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()