# 测试学生类
import unittest
class Student(object):
def __init__(self, name, score):
self.name = name
self.score = score
def get_grade(self):
if self.score>100 or self.score<0:
raise ValueError('成绩不合格')
elif self.score >= 80:
return 'A'
elif self.score >= 60:
return 'B'
else:
return 'C'
# 测试类
class TestStudent(unittest.TestCase):
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_70_to_100(self):
s1 = Student('Bart', 89)
s2 = Student('Lisa', 79)
self.assertEqual(s1.get_grade(), 'A')
self.assertEqual(s2.get_grade(), 'B')
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()
def setUp(self):
print('开始执行代码')#会在每一个方法开始执行
def tearDown(self):
print('结束执行代码')#会在每一个方法结束执行
if __name__ == '__main__':
unittest.main()
编写单元测试时,我们需要编写一个测试类,从unittest.TestCase继承。
以test开头的方法就是测试方法,不以test开头的方法不被认为是测试方法,测试的时候不会被执行。
unittest单元测试实例
最新推荐文章于 2022-11-19 22:51:57 发布