《Python编程-从入门到实践》第十一章习题训练

最新推荐文章于 2020-11-01 11:58:54 发布

a199805045

最新推荐文章于 2020-11-01 11:58:54 发布

阅读量255

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/a199805045/article/details/79952072

本文介绍了Python中如何使用unittest模块进行单元测试，并通过两个示例进行说明：一是测试城市和国家格式化输出的功能；二是测试员工类的薪资调整功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

11-1城市和国家：

def city_function(city, country):
    r = city.title() + ', ' + country.title()
    return r

import unittest
from city_functions import city_function


class NameTestCase(unittest.TestCase):

    def test_city_country(self):
        test_case = city_function('santiago', 'chile')
        self.assertEqual(test_case, 'Santiago, Chile')


unittest.main()

11-3雇员：

class 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, r = 5000):
        self.salary += r

import unittest
from employee import Employee

class TestEmployee(unittest.TestCase):

    def setUp(self):
        first_name = 'Daniel'
        last_name = 'Chung'
        salary = 10000
        self.myEmployee = Employee(first_name, last_name, salary)
        self.default_raise = 15000
        self.custom_raise = 20000

    def test_give_default_raise(self):
        self.myEmployee.give_raise()
        self.assertEqual(self.default_raise, self.myEmployee.salary)

    def test_give_custom_raise(self):
        self.myEmployee.give_raise(10000)
        self.assertEqual(self.custom_raise, self.myEmployee.salary)

unittest.main()