python测试类和setUp方法的简单处理

最新推荐文章于 2024-09-05 08:00:00 发布

文远先生

最新推荐文章于 2024-09-05 08:00:00 发布

阅读量766

点赞数

CC 4.0 BY-SA版权

分类专栏： python

本文链接：https://blog.youkuaiyun.com/qq_41633726/article/details/94389000

python 专栏收录该内容

14 篇文章

订阅专栏

本文通过实例演示如何使用python的unittest模块进行单元测试，重点介绍了如何测试类中的方法是否按预期工作，包括默认参数和自定义参数的情况。

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

python在unittest.TestCase类中有很多断言方法，具体就不列举了。
先写出需要测试的类

class Employee():
    def __init__(self,first,last,money):
        #定义了类中有哪些属性
        self.first=first
        self.last=last
        self.money=money
    def give_raise(self,add_money=5000):
        #如果增加的钱是默认的，那就增加5000，如果不是那就需要自己输入增加再增加
        self.money=add_money+self.money

然后测试如果加的钱是默认的是否只加5000，如果不是默认的而是自己输入的某一个值，是否增加了这个值的数额。

import unittest
from employee import Employee
#导入unittest模块和要测试的类

class TestEmployee(unittest.TestCase):
#用常用的断言方法来判断是否正确
    def setUp(self):
        self.employee_one=Employee('one','one',100000)
        self.true_money=[110000,105000]
    def test_give_default_raise(self):
        #测试类中的不按默认增加是否正确
        self.employee_one.give_raise(add_money=10000)
        self.assertIn(self.employee_one.money,self.true_money)
    def test_give_custom_raise(self):
        #测试类中的按默认增加5000是否正确
        self.employee_one.give_raise()
        self.assertIn(self.employee_one.money,self.true_money)
unittest.main()

这里我用的是setUp方法，需要创建一个调查对象（self.employee_one），还需要创建一个答案列表，之后测试检查是否程序给出的结果是符合答案的