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()