創建一個名爲test_cities.py的程序,對剛編寫的函數進行測試(別忘了,你需要導入模塊unittest 以及要測試的函數)。編寫一個名爲test_city_country() 的方法,覈實使用類似於'santiago' 和'chile' 這樣的值來調用前述函數時,得到的字符串是正確的。運行test_cities.py ,確認測試test_city_country() 通過了。
"""
city_functions.py
"""
def cityfunctions(City,Country):
full_name=City.title()+','+Country.title()
return full_name
"""
test_cites.py
"""
import unittest
from city_functions import cityfunctions
class CityTestCase(unittest.TestCase):
def test_city_country(self):
ans=cityfunctions('santiago','chile')
self.assertEqual(ans,'Santiago,Chile')
unittest.main()
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() 通過了。
"""
city_functions.py
"""
def cityfunctions(City,Country,population):
full_name=City.title()+','+Country.title()+'-'+str(population)
return full_name
"""
city_functions.py
"""
def cityfunctions(City,Country,population=0):
if population:
full_name=City.title()+','+Country.title()+'-'+str(population)
else:
full_name=City.title()+','+Country.title()
return full_name
def cityfunctions(City,Country,population=0):
if population:
full_name=City.title()+','+Country.title()+'-population '+str(population)
else:
full_name=City.title()+','+Country.title()
return full_name
"""
test_cites.py
"""
import unittest
from city_functions import cityfunctions
class CityTestCase(unittest.TestCase):
def test_city_country(self):
ans=cityfunctions('santiago','chile')
self.assertEqual(ans,'Santiago,Chile')
def test_city_country_population(self):
ans=cityfunctions('santiago','chile',population=5000000)
self.assertEqual(ans,'Santiago,Chile-population 5000000')
unittest.main()
11-3 僱員 :編寫一個名爲Employee 的類,其方法__init__() 接受名、姓和年薪,並將它們都存儲在屬性中。編寫一個名爲give_raise() 的方法,它默認將年薪增加5000美元,但也能夠接受其他的年薪增加量。
爲Employee 編寫一個測試用例,其中包含兩個測試方法:test_give_default_raise() 和test_give_custom_raise() 。使用方法setUp() ,以免在每個測試方法中都創建新的僱員實例。運行這個測試用例,確認兩個測試都通過了。
class Employee():
def _init_(self,first_name,last_name,salary):
self.first_name=first_name
self.lat_name=last_name
self.salary=salary
def give_raise(self,num=5000):
self.salary+=num
import unittest
from employee import Empolyee
class EmployeeTestCase(unittest.TestCase):
def setUp(self):
self.my_employee=empolyee('LeBron','James',2500)
def test_give_default_raise(self):
self.my_empolyee.give_raise()
self.assertEqual(self.my_empolyee.salary,7500)
def test_give_custom_raise(self):
self.my_empolyee.give_raise(3000)
self.assertEqual(self.my_empolyee.salary,5500)
unittest.main()