Homework 6.1 Chapter 11 Homework Exercises

11-1 Cities and countries
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 Population
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()