测试代码的练习——python编程从入门到实践

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

---

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

test_cities.py

---

import unittest
from city_functions import city_country

class CityFunctionsTestCase(unittest.TestCase):
    """测试city_functions.py"""

    def test_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 in 0.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

---

def city_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=''):
    if population:
        return city.title() + ', ' + country.title() + ' - population ' + str(population)
    else:
        return city.title() + ', ' + country.title()

　　修改函数 -> 运行test_cities.py -> 确认测试test_city_country()又通过了。

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