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

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