《Python编程：从入门到实践》习题答案——第11章 测试代码

# 11-1 城市和国家

# city_functions.py
def city_country_func(city, country, population = ''):
    """生成城市和国家"""
    if population:
        str = city + ' ' + country + ' ' + 'population=' + population
    else:
        str = city + "，" + country
    return str

from city_functions import city_country_func
import unittest

class CityTestCase(unittest.TestCase):
    """测试city_functions.py"""
    def test_city_country(self):
        formatted_city_country = city_country_func('Beijing', 'China')
        self.assertEqual(formatted_city_country, "Beijing，China")

# 11-2 人口数量
from city_functions import city_country_func
import unittest

class CityTestCase(unittest.TestCase):
    """测试city_functions.py"""
    def test_city_country(self):
        formatted_city_country = city_country_func('Beijing', 'China')
        self.assertEqual(formatted_city_country, "Beijing，China")

    def test_city_country_population(self):
        formatted_city_country_population = city_country_func('santiago','chile','5000000')
        self.assertEqual(formatted_city_country_population, 'santiago chile population=5000000')

# 11-3 雇员

# employee.py
class Employee(object):
    """decstring for Employee"""
    def __init__(self, first_name, last_name, salary):
        self.first_name = first_name
        self.last_name = last_name
        self.salary = int(salary)
    def give_raise(self, increase = ''):
        if increase != '':
            self.salary += ''
        else:
            self.salary += 5000

import unittest
from employee import Employee

class TestEmployee(unittest.TestCase):
    def setUp(self):
        self.employee = Employee('LeBron', 'james', 15000)

    def test_give_default_raise(self):
        self.test_employee.give_raise(10000)

    def test_give_custom_raise(self):
        self.test_employee.give_raise()

unittest.main()

class TestEmployee(unittest.TestCase):
    def setUp(self):
        self.my_employee = Employee("郑超", "23", 5000)

    def test_give_default_raise(self):
        self.my_employee.give_raise()
        self.assertEqual(10000, self.my_employee.money)

    def test_give_custom_raise(self):
        self.my_employee.give_raise(10000)
        self.assertEqual(15000, self.my_employee.money)

if __name__ == '__main__':
    unittest.main()