第六周作业（11章）

11-1

def city_functions(city, country):
	msg = city.capitalize() + ', ' + country.capitalize()
	return msg

#coding:gbk
import unittest
from city import city_functions

class CityTestCase(unittest.TestCase):
	'''测试city_functions.py'''
	
	def test_city_country(self):
		city_country_name = city_functions('shanghai', 'china')
		self.assertEqual(city_country_name, 'Shanghai, China')

unittest.main()

11-2

def city_functions(city, country, population):
	msg = city.capitalize() + ', ' + country.capitalize()
	return msg + ' - population ' + str(population)

def city_functions(city, country, population=''):
	msg = city.capitalize() + ', ' + country.capitalize()
	if population:
		return msg + ' - population ' + str(population)
	else:
		return msg

11-3

class Employee():
	def __init__(self, first_name, last_name, annual_pay):
		self.first_name = first_name
		self.last_name = last_name
		self.annual_pay = int(annual_pay)
	
	def give_raise(self, incease = 5000):
		return self.annual_pay + incease
	 

#coding:gbk

import unittest
from employee import Employee

class TestEmployee(unittest.TestCase):
	def setUp(self):
		self.my_employee = Employee('kiana', 'Kaslana', 2666666)
		
	def test_give_default_raise(self):
		my_pay = self.my_employee.give_raise()
		self.assertEqual(my_pay, self.my_employee.annual_pay + 5000)
	
	def  test_give_custom_raise(self):
		my_pay = self.my_employee.give_raise(5613)
		self.assertEqual(my_pay, self.my_employee.annual_pay + 5613)
		
unittest.main()