第十一章 动手试一试

11-1 城市和国家：编写一个函数，它接受两个形参：一个城市名和一个国家名。这 个函数返回一个格式为 City, Country 的字符串，如 Santiago, Chile。将这个函数存储 在一个名为 city _functions.py的模块中。 创建一个名为 test_cities.py的程序，对刚编写的函数进行测试（别忘了，你需要导 入模块 unittest 以及要测试的函数）。编写一个名为 test_city_country()的方法，核实使用类似于'santiago'和'chile'这样的值来调用前述函数时，得到的字符串是正确的。 运行 test_cities.py，确认测试 test_city_country()通过了。

#city_functions.py
def get_city_country(city, country):
	return city.title() + ',' + country.title()

#test_cities.py
import unittest
from city_functions import get_city_country 
class city_test(unittest.TestCase):
	def test_city_country(self):
		answer = get_city_country('santiago', 'chile')
		self.assertEqual(answer, 'Santiago,Chile')

unittest.main()

11-3 雇员：编写一个名为 Employee 的类，其方法__init__()接受名、姓和年薪，并 将它们都存储在属性中。编写一个名为 give_raise()的方法，它默认将年薪增加 5000 美元，但也能够接受其他的年薪增加量。 为 Employee 编写一个测试用例，其中包含两个测试方法：test_give_default_ raise()和 test_give_custom_raise()。使用方法 setUp()，以免在每个测试方法中都创 建新的雇员实例。运行这个测试用例，确认两个测试都通过了。

#employee.py
class Employee():
	def __init__(self, first_name, last_name, salary):
		self.first_name = first_name
		self.last_name = last_name
		self.salary = salary

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

#test_employee.py
import unittest
from employee import Employee
class TestEmployee(unittest.TestCase):
	def setUp(self):
		self.test_employee = Employee('deyi', 'chen', 25000)

	def test_give_default_raise(self):
		self.test_employee.give_raise()
		self.assertEqual(self.test_employee.salary, 30000)

	def test_give_custom_raise(self):
		self.test_employee.give_raise(10000)
		self.assertEqual(self.test_employee.salary, 35000)

unittest.main()