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 city_country(city, country):
return city + ', ' + country
#test_cities.py
import unittest
from city_functions import city_country
class my_test(unittest.TestCase):
def test_city_country(self):
res = city_country('Guangzhou', 'China')
self.assertEqual(res, 'Guangzhou, China')
unittest.main()
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
>>>
11-3 雇员
编写一个名为Employee 的类, 其方法__init__() 接受名、 姓和年薪, 并将它们都存储在属性中. 编写一个名为give_raise() 的方法, 它默认将年薪增加5000美元, 但也能够接受其他的年薪增加量.
为Employee 编写一个测试用例, 其中包含两个测试方法: test_give_default_raise() 和test_give_custom_raise() . 使用方法setUp() , 以免在每个测试方法中都创建新的雇员实例. 运行这个测试用例, 确认两个测试都通过了.
class Employee():
def __init__(self, first, last, salary):
self.firstname, self.lastname, self.salary = first, last, salary
def give_raise(self, inc = 5000):
self.salary += inc
return self.salary
import unittest
class my_test(unittest.TestCase):
def setUp(self):
self.guy = Employee('Marc', 'Levy', 50000)
def test_give_default_raise(self):
res = self.guy.give_raise()
self.assertEqual(res, 55000)
def test_give_custom_raise(self):
res = self.guy.give_raise(50000)
self.assertEqual(res, 100000)
unittest.main()
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
>>>