Python单元测试

python 复制代码
import unittest #必须要导入单元测试的包


class Student(object):
    def __init__(self, name, score):
        self.name = name
        self.score = score
    def get_grade(self):
        if self.score > 100:
            #返回错误不能用return,应该用raise 
            raise ValueError("成绩不能大于100")
        if self.score >= 80:
            return 'A'
        if self.score >= 60:
            return 'B'
        if self.score < 0:
            raise ValueError("成绩不能为0")
        return 'C'
    
class TestStudent(unittest.TestCase): # 测试类必须要继承这个包
    # 这两个方法会在每一个测试用例的开始和结束都执行
    def setUp(self):
        print('setUp...')

    def tearDown(self):
        print('tearDown...')
    # 测试方法前面必须要加test_
    def test_80_to_100(self):
        s1 = Student('Bart', 80)
        s2 = Student('Lisa', 100)
        self.assertEqual(s1.get_grade(), 'A')
        self.assertEqual(s2.get_grade(), 'A')
    
    def test_60_to_80(self):
        s1 = Student('Bart', 60)
        s2 = Student('Lisa', 79)
        self.assertEqual(s1.get_grade(), 'B')
        self.assertEqual(s2.get_grade(), 'B')

    def test_0_to_60(self):
        s1 = Student('Bart', 0)
        s2 = Student('Lisa', 59)
        self.assertEqual(s1.get_grade(), 'C')
        self.assertEqual(s2.get_grade(), 'C')

    def test_invalid(self):
        s1 = Student('Bart', -1)
        s2 = Student('Lisa', 101)
        with self.assertRaises(ValueError):
            s1.get_grade()
        with self.assertRaises(ValueError):
            s2.get_grade()

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

写单元测试的时候,不用考虑之前写的代码的逻辑,直接自己编写测试用例就行

单元测试的测试用例要覆盖常用的输入组合、边界条件和异常

参考文章:单元测试 - 廖雪峰的官方网站

问题一:上面的单元测试中有四方测试,如果只测试其中的一个呢? 再进一步,如何一步一步的调试呢?

命令行方式

python 复制代码
python3 -m unittest -k test_80_to_100 TestStudent.py

但是这样每次都要输入这么长的命令,如果通过IDE直接选中这个函数进行调试呢?

我们先在python插件中设置一下

配置文件也设置一下

安装完成之后,点击这个类似于烧杯的图标

问题二:

那单元测试的难点在于写测试用例呀,测试用例需要自己想,而且还不能出错了,那只能写一些比较简单的测试用例,用人脑可以简单看出来的。 有个疑问是:算法比赛中的那些测试用例是怎么搞出来的?

相关推荐
HAH-HAH8 小时前
【Python 入门】(2)Python 语言基础(变量)
开发语言·python·学习·青少年编程·个人开发·变量·python 语法
SunnyDays10119 小时前
Python 轻松实现替换或修改 PDF 文字
python·替换pdf文字·修改pdf·修改pdf文字
Just_Paranoid10 小时前
【Settings】恢复出厂设置密码校验
android·python·settings·sha256·hmac-sha256
西猫雷婶11 小时前
pytorch基本运算-Python控制流梯度运算
人工智能·pytorch·python·深度学习·神经网络·机器学习
子午11 小时前
Python的uv包管理工具使用
开发语言·python·uv
java1234_小锋12 小时前
Scikit-learn Python机器学习 - 分类算法 - 朴素贝叶斯
python·机器学习·scikit-learn
凡梦千华12 小时前
Django时区感知
后端·python·django
程序猿 小项目大搞头12 小时前
即梦批量生成图片软件使用运营大管家-即梦图片批量生成器
python
dpxiaolong13 小时前
RK3588 Android12默认移除导航栏
开发语言·python
max50060013 小时前
基于多元线性回归、随机森林与神经网络的农作物元素含量预测及SHAP贡献量分析
人工智能·python·深度学习·神经网络·随机森林·线性回归·transformer