引入其他类

我们在测试的过程中，不是所有时候都要把代码和测试代码放到一起的，比如开发的代码很长很长，那么我们如果把开发的代码全部拷贝出来，这样整个文件就显得比较冗余。不方便检查。我们可以使用import的方法把待测试的代码引入到我们的测试文件中。比如这样一段代码，单独保存到一个文件中，我保存的文件名是testclass.py

# -*- coding: utf-8 -*-
class testnum():

def sumnum(self,a,b):
return a + b

def delnum(self,a,b):
return a - b

def hello(self):
return "hello world"

def chengfa(self,a,b):
return a * b

要针对这段代码做测试，要在我们的测试代码中先引入这个类

from testclass import testnum

然后在我们的测试代码中的setUp函数中把这个类实例化，在测试的函数中就可以调用这个类了。完整代码如下：

# -*- coding: utf-8 -*-
import unittest
from testclass import testnum
class testNum(unittest.TestCase):
def setUp(self):
self.testmodel = testnum()
def tearDown(self):
pass
def testSum(self):
self.assertEqual(2,self.testmodel.sumnum(1,1))
self.assertEqual(0,self.testmodel.delnum(1,1))
if __name__ == '__main__':
suite1 = unittest.TestLoader().loadTestsFromTestCase(testNum)
unittest.TextTestRunner(verbosity=2).run(suite1)

被测试文件：

# -*- coding: utf-8 -*-
class testnum():

def sumnum(self,a,b):
return a + b

def delnum(self,a,b):
return a - b

def hello(self):
return "hello world"

def chengfa(self,a,b):
return a * b
if __name__ == '__main__':
a = testnum()
print a.sumnum(1,2)