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()通过了。

def city_country(city_name, country_name):
    str = city_name + ", " + country_name
    return str.title()

import unittest
from city_functions import city_country

class CityTestCase(unittest.TestCase):
    def test_city_country(self):
        """测试是否能连接两个字符串"""
        str = city_country('shanghai', 'china')
        self.assertEqual(str, 'Shanghai, China')

if __name__ == '__main__':
    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_country_population()通过了。

def city_country(city_name, country_name, population=''):
    if population:
        str = city_name.title() + ", " + country_name.title() + " - " + population
    else:
        str = city_name.title() + ", " + country_name.title()
    return str

import unittest
from city_functions import city_country

class CityTestCase(unittest.TestCase):
    def test_city_country(self):
        """测试是否能连接两个字符串"""
        str = city_country('shanghai', 'china')
        self.assertEqual(str, 'Shanghai, China')

    def test_city_country_population(self):
        """测试连接三个字符串"""
        str = city_country('santigo', 'chile', 'population=500000')
        self.assertEqual(str, 'Santigo, Chile - population=500000')

if __name__ == '__main__':
    unittest.main()