高階程式設計技術(Python)作業11
11-1 城市和國家:編寫一個函式,它接受兩個形參:一個城市名和一個國家名。這個函式返回一個格式為City, Country的字串,如Santiago, Chile 。將這個函式儲存在一個名為city_functions.py的模組中。建立一個名為test_cities.py的程式,對剛編寫的函式進行測試(別忘了, 你需要匯入模組unittest 以及要測試的函式)。編寫一個名為test_city_country() 的方法,核實使用類似於’santiago’和’chile’這樣的值來呼叫前述函式時,得到的字串是正確的。執行test_cities.py,確認測試test_city_country()通過了。
Solution:
#city_functions.py
def city_country(City, Country):
output = City.title() +', ' + Country.title()
return output
#test_cities.py
import unittest
from city_functions import city_country
class CityTestCase(unittest.TestCase):
"""test city_functions.py"""
def test_city_country (self):
formatted_output = city_country('santiago', 'chile')
self.assertEqual(formatted_output, 'Santiago, Chile')
unittest.main()
Output:
.
----------------------------------------------------------------------
Ran 1 test in 0.001s
OK
11-2 人口數量:修改前面的函式,使其包含可選的形參population ,並返回一個格式為City, Country - population xxx 的字串,如Santiago, Chile - population 5000000 。
執行test_cities.py,確認測試test_city_country()通過了。
再編寫一個名為test_city_country_population() 的測試,核實可以使用類似於’santiago’ 、 ‘chile’ 和’population=5000000’ 這樣的值來呼叫這個函式。再次執行test_cities.py,確認測試test_city_country_population()通過了。
Solution:
#city_functions.py
def city_country(City, Country, population=''):
if population:
output = City.title() +', ' + Country.title() + ' - population ' + str(population)
else:
output = City.title() +', ' + Country.title()
return output
#test_cities.py
import unittest
from city_functions import city_country
class CityTestCase(unittest.TestCase):
"""test city_functions.py"""
def test_city_country(self):
formatted_output = city_country('santiago', 'chile')
self.assertEqual(formatted_output, 'Santiago, Chile')
def test_city_country_population(self):
formatted_output = city_country('santiago', 'chile', 5000000)
self.assertEqual(formatted_output, 'Santiago, Chile - population 5000000')
unittest.main()
Output:
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
11-3 僱員:編寫一個名為Employee 的類,其方法init() 接受名、姓和年薪,並將它們都儲存在屬性中。編寫一個名為give_raise()的方法,它預設將年薪增加5000美元,但也能夠接受其他的年薪增加量。
為Employee 編寫一個測試用例,其中包含兩個測試方法:test_give_default_raise() 和test_give_custom_raise() 。使用方法setUp() ,以免在每個測試方法中都建立新的僱員例項。執行這個測試用例,確認兩個測試都通過了。
Solution:
#employee_class.py
class Employee():
"""build a record for a employee"""
def __init__(self, first_name, last_name, salary):
"""initial a employee"""
self.first_name = first_name
self.last_name = last_name
self.salary = salary
def give_raise(self, money=5000):
"""giving raise"""
self.salary += money
#test_employee_class.py
import unittest
from employee_class import Employee
class TestEmployeeClass(unittest.TestCase):
"""test Employee"""
def setUp(self):
"""set up a test sample"""
self.test_employee = Employee("Tim", "Ashero", 5000)
def test_give_default_raise(self):
"""test default raise"""
self.test_employee.give_raise()
test_salary = self.test_employee.salary
self.assertEqual(test_salary, 10000)
def test_give_custom_raise(self):
"""test custom raise"""
self.test_employee.give_raise(8000)
test_salary = self.test_employee.salary
self.assertEqual(test_salary, 13000)
unittest.main()
Output:
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK