- 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()
写单元测试的时候,不用考虑之前写的代码的逻辑,直接自己编写测试用例就行
单元测试的测试用例要覆盖常用的输入组合、边界条件和异常
参考文章:单元测试 - 廖雪峰的官方网站
命令行方式
python3 -m unittest -k test_80_to_100 TestStudent.py
但是这样每次都要输入这么长的命令,如果通过IDE直接选中这个函数进行调试呢?

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

配置文件也设置一下

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

那单元测试的难点在于写测试用例呀,测试用例需要自己想,而且还不能出错了,那只能写一些比较简单的测试用例,用人脑可以简单看出来的。 有个疑问是:算法比赛中的那些测试用例是怎么搞出来的?
问题三:
单元调试过程中,如何强行中断调试?
问题四:
在ros环境中, 有时候需要source一下包,但是调试无法source