第十一章

11-1

#city_functions.py
def city_and_country(city, country):
	string = city.title() + ', ' + country.title()
	return string

#test_cities.py
import unittest
from city_functions import city_and_country

class cityTestCase(unittest.TestCase):
	def test_city_country(self):
		string = city_and_country('santiago', 'chile')
		self.assertEqual(string, 'Santiago, Chile')

unittest.main()

11-2

#city_functions.py
def city_and_country(city, country, population = '0'):
	string = city.title() + ', ' + country.title()
	if population != '0':
		string = string + ' - population ' + population + ' '
	return string

#test_cities.py
import unittest
from city_functions import city_and_country

class cityTestCase(unittest.TestCase):
	def test_city_country(self):
		string = city_and_country('santiago', 'chile', '5000000')
		self.assertEqual(string, 'Santiago, Chile - population 5000000 ')

unittest.main()

11-3

import unittest

class Employee():
	def __init__(self, firstName, lastName, salary):
		self.firstName = firstName
		self.lastName = lastName
		self.salary = salary
		
	def give_raise(self, increasement = 5000):
		self.salary = self.salary + increasement
		
class TestEmployeeCase(unittest.TestCase):
	def setUp(self):
		self.employee = Employee('Zilin', 'Huang', 10000)
	
	def test_give_default_raise(self):
		self.employee.give_raise()
		self.assertEqual(self.employee.salary, 15000)
		
	def test_give_custom_raise(self):
		self.employee.give_raise(10000)
		self.assertEqual(self.employee.salary, 20000)
		
unittest.main()