# -*- coding: utf-8 -*-
import unittest
class Student(object):
def __init__(self, name, score):
self.name = name
self.score = score
def get_grade(self):
if self.score >= 80 and self.score <= 100:
return 'A'
elif self.score >= 60 and self.score <80:
return 'B'
elif self.score < 60 and self.score >=0:
return 'C'
else:
raise ValueError
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_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()
单元测试例题解析:
依吾拙见,写成了这个样子,原来的错误是这样的
class Student(object):
def __init__(self, name, score):
self.name = name
self.score = score
def get_grade(self):
if self.score >= 60:
return 'B'
if self.score >= 80:
return 'A'
return 'C'
重新划分了一下范围,结合单元测试中已有的ValueError
改写类的get_grade
方法就成了,这应该是最简单的方法了,一定还有更厉害一点的写法。
最近在同时学Abaqus和Python,想着用Python做下Abaqus的二次开发,结合自己的模态试验,兴许能往振动疲劳方向走一走