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
defcity_country(city, country):return city.title() + ',' + country.title()
test_cities.py
importunittestfrom city_functions importcity_countryclassCityFunctionsTestCase(unittest.TestCase):"""测试city_functions.py"""
deftest_city_country(self):"""测试函数能够返回这样的字符串:Santiago, Chile"""c_c= city_country('santiago', 'chile')
self.assertEqual(c_c,'Santiago, Chile')if __name__ == "__main__":
unittest.main()
运行test_cities.py:
Ran 1 test in0.002s
OK
经确认测试test_city_country()通过了。
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
defcity_country(city, country, population):return city.title() + '.' + country.title() + '- population' + str(population)
修改函数 -> 运行test_cities.py -> 确认测试test_city_country()未通过。
city_functions.py
def city_country(city, country, population=''):ifpopulation:return city.title() + ',' + country.title() + '- population' +str(population)else:return city.title() + ',' + country.title()
修改函数 -> 运行test_cities.py -> 确认测试test_city_country()又通过了。
deftest_city_country_population(self):"""测试函数能返回这样的字符串:Santiago,Chile - population 5000000"""c_c_p= city_country('santiago', 'chile', population=5000000)
self.assertEqual(c_c_p,'Santiago, Chile - population 5000000')
在test_cities.py中添加一个名为test_city_country_population()的测试 -> 再次运行test_cities.py -> 确认测试test_city_country_population()通过了。