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_fucntions.py
def city_country(city, country):
return city.title() + ', ' + country.title()
# test_cities.py
import unittest
from city_functions import city_country
class CitiesTestCase(unittest.TestCase):
"""Test for 'city_functions.py'."""
def test_city_country(self):
"""Do cites like 'Santiago, Chile' work?"""
formatted_city_name = city_country('santiago', 'chile')
self.assertEqual(formatted_city_name, 'Santiago, Chile')
unittest.main()
运行结果
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
11-2 人口数量:修改前面的函数,使其包含第三个必不可少的形参 population,并返回一个格式为 City, Country – population xxx 的字符串,如 Santiago, Chile – population 5000000。运行 test_cities.py,确认测试 test_city_country()未通过。
修改上述函数,将形参 population 设置为可选的。再次运行 test_cities.py,确认测试 test_city_country()又通过了。
再编写一个名为 test_city_country_population()的测试,核实可以使用类似于’santiago’ 、’chile’和’population=5000000’这样的值来调用这个函数。再次运行test_cities.py,确认测试 test_city_country_population()通过了。
代码1
# city_fucntions.py
def city_country(city, country, population=''):
return city.title() + ', ' + country.title() + ' - '\
'population ' + population
运行结果1
F
======================================================================
FAIL: test_city_country (__main__.CitiesTestCase)
Do cites like 'Santiago, Chile - population xxx' work?
----------------------------------------------------------------------
Traceback (most recent call last):
File "test_cities.py", line 12, in test_city_country
self.assertEqual(formatted_city_name, 'Santiago, Chile')
AssertionError: 'Santiago, Chile - population ' != 'Santiago, Chile'
- Santiago, Chile - population
+ Santiago, Chile
----------------------------------------------------------------------
Ran 1 test in 0.000s
FAILED (failures=1)
代码2
# city_fucntions.py
def city_country(city, country, population=''):
if population:
return city.title() + ', ' + country.title() + ' - '\
'population ' + population
else:
return city.title() + ', ' + country.title()
运行结果2
.
----------------------------------------------------------------------
Ran 1 test in 0.001s
OK
代码3
# test_cities.py
...
def test_city_country(self):
formatted_city_name = city_country('santiago', 'chile', '5000000')
self.assertEqual(formatted_city_name, 'Santiago, Chile - population 5000000')
...
运行结果3
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
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, annual_salary):
self.first_name = first_name
self.last_name = last_name
self.annual_salary = int(annual_salary)
def give_raise(self, incre=50000):
if incre == 50000:
self.annual_salary += 50000
return self.annual_salary
else:
self.annual_salary += incre
return self.annual_salary
# test_employee.py
import unittest
from employee import Employee
class EmployeeTestCase(unittest.TestCase):
def setUp(self):
self.emp = Employee('Hua', 'Li', 50000)
def test_give_default_raise(self):
default = self.emp.give_raise()
self.assertEqual(default, 50000+50000)
def test_give_custom_raise(self):
custom = self.emp.give_raise(20000)
self.assertEqual(custom, 50000+20000)
unittest.main()
运行结果
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK