《Python编程从入门到实践》Chapter 11练习题

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

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

#先编写函数并存储
def get_city_country(city,country):
    """返回城市及国家"""

    city = input("City: ")
    country = input("Country: ")
    city_country = city.title() + "," + country.title()
    return city_country

#编写测试
import unittest
from city_functions import get_city_country


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

    def test__city_country(self):
        """测试函数能否正确处理像'Santiago, Chile'这样的字符串 """
        my_city_country = get_city_country('santiago', "chile")
        self.assertEqual(my_city_country, 'Santiago,Chile')


unittest.main()

city_functions.py

def get_city_country(city,country):
    """返回城市及国家"""

    city = input("City: ")
    country = input("Country: ")
    city_country = city.title() + "," + country.title()
    return city_country

虽然运行这个程序是有输入的步骤，但是因为结果已经定了’Santiago,Chile’，为了让测试通过和有意义，必须要输入’santiago’和 “chile”。否则就会导致断言失败。
这里只要知道，在测试过程中可以采用input()函数输入。而具体的使用方法，还需要以后根据情况编写合适的代码来使用。