11-1 城市和国家
city_functions.py

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

test_cities.py

import unittest
from city_functions import get_formatted_city

class CitesTestCase(unittest.TestCase):
    def test_city_country(self):
        formatted_city = get_formatted_city('santiago', 'chile')
        self.assertEqual(formatted_city, 'Santiago, Chile')

unittest.main()

11-2 人口数量
city_functions.py

def get_formatted_city(city, country, population=''):
    if population:
        return (city + ', ' + country).title() + ' - population ' + str(population)
    else:
        return (city + ', ' + country).title() 

test_cities.py

import unittest
from city_functions import get_formatted_city

class CitesTestCase(unittest.TestCase):
    def test_city_country(self):
        formatted_city = get_formatted_city('santiago', 'chile')
        self.assertEqual(formatted_city, 'Santiago, Chile')
    def test_city_country_population(self):
        formatted_city = get_formatted_city('santiago', 'chile', population = 5000000)
        self.assertEqual(formatted_city, 'Santiago, Chile - population 5000000')

unittest.main()