Python 函数练习：城市和国家信息格式化

Python 函数练习：城市和国家信息格式化

本教程将演示如何使用 Python 函数来格式化城市和国家信息，并使用单元测试进行验证。

1. 编写函数 city_country()

首先，创建一个名为 city_functions.py 的文件，并在其中定义一个名为 city_country() 的函数。该函数接受两个参数：城市名和国家名。函数应该返回一个格式化的字符串，例如 'Santiago, Chile'。

def city_country(city, country):
    return f'{city.title()}, {country.title()}'

2. 编写测试代码

接下来，创建一个名为 test_cities.py 的文件，并在其中编写测试代码。这个文件将使用 unittest 模块来测试 city_country() 函数。

import unittest
from city_functions import city_country

class CityCountryTestCase(unittest.TestCase):
    def test_city_country(self):
        formatted_name = city_country('santiago', 'chile')
        self.assertEqual(formatted_name, 'Santiago, Chile')

if __name__ == '__main__':
    unittest.main()

3. 运行测试

在命令行中运行 test_cities.py 文件。如果测试通过，您应该看到类似以下的输出：

....
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

代码解释

• city_functions.py 文件
  • city_country() 函数使用 f-string 格式化字符串，并将城市和国家名首字母大写。
• test_cities.py 文件
  • CityCountryTestCase 类继承自 unittest.TestCase 类，并定义了一个名为 test_city_country() 的测试方法。
  • test_city_country() 方法使用 assertEqual() 断言，检查 city_country() 函数返回的格式化字符串是否与预期结果一致。

总结

本教程演示了如何使用 Python 函数格式化城市和国家信息，并使用单元测试进行验证。您可以在此基础上进一步扩展代码，例如添加更多测试用例或实现其他功能。