《Python編程：從入門到實踐》課后習題及答案—第11章

 

第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() 通過了。

city_functions.py

def city_country(city_name, country_name):
    return city_name + " , " + country_name

test_cities.py

import unittest
from city_functions import city_country

class NameTestCase(unittest.TestCase):
    def test_city_country_1(self):
        name = city_country("santiago", "chile")
        self.assertEqual(name, "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() 通過了。

import unittest
from city_functions import city_country

class NameTestCase(unittest.TestCase):
    def test_city_country(self):
        name = city_country("santiago", "chile")
        self.assertEqual(name, "santiago , chile")
    def test_city_country_population(self):
        name = city_country("santiago", "chile","50000")
        self.assertEqual(name, "santiago , chile-50000")

unittest.main()

 11-3 雇員 ：編寫一個名為Employee 的類，其方法__init__() 接受名、姓和年薪，並將它們都存儲在屬性中。編寫一個名為give_raise() 的方法，它默認將年薪增加5000美元，但也能夠接受其他的年薪增加量。

為Employee 編寫一個測試用例，其中包含兩個測試方法：test_give_default_raise() 和test_give_custom_raise() 。使用方法setUp() ，以免在每個測試方法中都創建新的雇員實例。運行這個測試用例，確認兩個測試都通過了。

import unittest

class Employee():
    def __init__(self, first, last, pay):
        self.first = first
        self.last = last
        self.pay = pay

    def give_raise(self, raise_pay = 5000):
        self.pay += raise_pay

class TestEmployee(unittest.TestCase):
    def setUp(self):
        self.test_employee = Employee("john", "smith", 8000)

    def test_give_default_raise(self):
        self.test_employee.give_raise()
        self.assertEqual(13000, self.test_employee.pay)

    def test_give_custom_raise(self):
        self.test_employee.give_raise(12000)
        self.assertEqual(20000, self.test_employee.pay)

unittest.main()