Python编程：从入门到实践（课后习题11）

# 11-1 城市和国家
# city_functions.py
def city_country(city, country):
city = city
country = country
cc = city + ' ' + country
return cc.title()

# test_cities.py
import unittest
from city_functions import city_country

class NamesTestCase(unittest.TestCase):

def test_city_country(self):
formatted_city = city_country('santiago', 'chile')
self.assertEqual(formatted_city, 'Santiago Chile')

unittest.main()

# 11-2 人口数量
# city_functions.py
def city_country(city, country, population=''):
city = city
country = country
population = population
if population:
cc = city + ' ' + country + ' - population ' + str(population)
else:
cc = city + ' ' + country
return cc.title()

# test_cities.py
import unittest
from city_functions import city_country

class NamesTestCase(unittest.TestCase):

def test_city_country(self):
formatted_city = city_country('santiago', 'chile')
self.assertEqual(formatted_city, 'Santiago Chile')

def test_city_country_population(self):
formatted_population = city_country('santiago', 'chile', 5000000)
self.assertEqual(formatted_population, 'Santiago Chile - Population 5000000')

unittest.main()

# 11-3 雇员
# employee.py
class Employee():

def __init__(self, firstname, lastname, salary):
self.firstname = firstname
self.lastname = lastname
self.salary = salary

def give_raise(self, add_salary=5000):
self.salary += add_salary

# test_employee.py
import unittest
from employee import Employee

class EmployeeTestCase(unittest.TestCase):

def test_give_default_raise(self):
formatted_default = Employee('zhou', 'kai', 2000)
formatted_default.give_raise()
self.assertEqual(formatted_default.salary, 7000)

def test_give_custom_raise(self):
formatted_default = Employee('zhou', 'kai', 2000)
formatted_default.give_raise(8000)
self.assertEqual(formatted_default.salary, 10000)

unittest.main()

# setUp()方法
# test_employee.py
import unittest
from employee import Employee

class EmployeeTestCase(unittest.TestCase):

def setUp(self):
self.formatted_default = Employee('zhou', 'kai', 2000)

def test_give_default_raise(self):
self.formatted_default.give_raise()
self.assertEqual(formatted_default.salary, 7000)

def test_give_custom_raise(self):
self.formatted_default.give_raise(8000)
self.assertEqual(formatted_default.salary, 10000)

unittest.main()