11-1 城市和国家
city_functions.py
# 城市和国家
def get_city_country(city, country):
"""返回格式为City, Country的字符串"""
city_country = city + ', ' + country
return city_country.title()
test_cities.py
import unittest
from city_functions import get_city_country
class CitiesTestCase(unittest.TestCase):
"""测试get_city_country"""
def test_city_country(self):
city_country = get_city_country('santiago', 'chile')
self.assertEqual(city_country, 'Santiago, Chile')
unittest.main()
过程中出错点:
1.unittest.TestCase中的T和C都是大写,小写会报错调用失败。
2.get_city_country中的参数列表’satiago’是字符串形式,忘记加引号提示satiago未定义。
最终输出:
11-2 人口数量
修改前面的函数,使其包含第三个必不可少的形参population,并返回一个格式为City,Country - population xxx的字符串,如Santiago,Chile - population 5000000.
# 城市和国家
def get_city_country(city, country, population):
"""返回格式为City, Country的字符串"""
city_country = city + ', ' + country + ' - population ' + population
return city_country.title()
运行test_cities.py,确认测试test_city_country()未通过。
修改上述函数,将形参population设置为可选的。
# 城市和国家
def get_city_country(city, country, population = ''):
"""返回格式为City, Country的字符串"""
if population:
city_country = (city + ', ' + country).title() + ' - ' + population
else:
city_country = (city + ', ' + country).title()
return city_country
再次运行test_cities.py,确认测试test_city_country()又通过了。
再编写一个名为test_city_country_population()的测试,核实可以使用类似于’santiago’,'chile’和’population=5000000’这样的值来调用这个函数。
import unittest
from city_functions import get_city_country
class CitiesTestCase(unittest.TestCase):
"""测试get_city_country"""
def test_city_country(self):
city_country = get_city_country('santiago', 'chile')
self.assertEqual(city_country, 'Santiago, Chile')
def test_city_country_population(self):
city_country = get_city_country('santiago', 'chile',
'population 5000000')
self.assertEqual(city_country, 'Santiago, Chile - population 5000000')
unittest.main()
再次运行test_cities.py,确认测试test_city_country_population()通过了。