一、需求分析

需求:一个类 MyClass，有两个功能：add, sub

1.先功能设计

# myclass.py

class MyClass(object):

# 加法

def add(self):

pass

# 减法

def sub(self):

pass

2.再功能测试

import unittest

from myclass import MyClass

class mytest(unittest.TestCase):

# 初始化

def setUp(self):

self.mc = MyClass()

# 退出清理

def tearDown(self):

pass

# 测试加法

def test_add(self):

self.assertEqual(3, self.mc.add(1, 2))

# 测试减法

def test_sub(self):

self.assertEqual(1, self.mc.sub(2, 1))

if __name__ =='__main__':

unittest.main()

3.再功能实现

# myclass.py

class MyClass(object):

def __init__(self):

super().__init__()

def add(self, x, y):

return x + y

def sub(self, x, y):

return x-y

4.进行测试

$ python myclass_test.py

OK，现在可以向客户交货了。

老板发话，大家幸苦了，明天放假休息一天。

一切都很美好！

二、需求有变

等等，客户半夜给老板打电话，说有些东西要变一下。

说好的休一天假泡汤了

客户说，1+1=2没有问题，但如果是'1'+'1'='11'是不允许的

这尼玛什么变态的需求啊！好吧，客户是上帝

1.修改功能测试

import unittest

from myclass import MyClass

class mytest(unittest.TestCase):

# 初始化

def setUp(self):

self.mc = MyClass()

# 退出清理

def tearDown(self):

pass

# 测试加法

def test_add(self):

self.assertEqual(3, self.mc.add(1, 2))

self.assertRaises(ValueError, self.mc.add, '1', '2') # 两个数都不是数字，则需要报错

# 测试减法

def test_sub(self):

self.assertEqual(1, self.mc.sub(2, 1))

if __name__ =='__main__':

unittest.main()

2.修改功能实现

# myclass.py

class MyClass(object):

def __init__(self):

super().__init__()

def add(self, x, y):

#return x + y

number_types = (int, float, complex)

if isinstance(x, number_types) and isinstance(y, number_types):

return x + y

else:

raise ValueError

def sub(self, x, y):

return x-y

3.进行测试

$ python myclass_test.py

好了，交货！

求求您千万别再提什么别的变态的需求了。