代码测试（城市和国家）

11-1 城市和国家：编写一个函数，它接受两个形参：一个城市名和一个国家名。这个函数返回一个格式为City, Country的字符串，如Santiago, Chile。将这个函数存储在一个名为city_functions.py的模块中。

创建一个名为test_cities.py的程序，对刚编写的函数进行测试（别忘了，你需要导入模块unittest以及要测试的函数）。编写一个名为test_city_country()的方法，核实使用类似于’santiago’和’chile’这样的值来调用前述函数时，得到的字符串是正确的。运行test_cities.py，确认测试test_city_country()通过了。

1、city_functions.py

def city_country(city,country):
    full_msg = city+','+country
    return full_msg.title()

2、

import unittest
from city_functions import city_country #导入模块函数
class CityCountryTest(unittest.TestCase):#测试类
    def test_city_country(self): #函数在运行时自动调用
        full_msg = city_country('santigo','chile')
        self.assertEqual(full_msg,'Santigo,Chile')#断言判断结果
        
unittest.main()

3、结果

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

1、city_functions.py

def city_country(city,country,population=' '):
    if population != ' ':
        full_msg = city.title() +','+country.title()+'-population' +str(population)
    else:
        full_msg = city.title()+','+country.title()
    return full_msg

2

import unittest
from city_functions import city_country
class CityCountryTest(unittest.TestCase):
    def test_city_country(self):
        full_msg = city_country('santigo','chile')
        self.assertEqual(full_msg,'Santigo,Chile')
        
    def test_city_country_population(self):
        full_msg = city_country('santigo','chile',555050)
        self.assertEqual(full_msg,'Santigo,Chile-population 555050')    
unittest.main()

3、结果