引入其他类

我们在测试的过程中，不是所有时候都要把代码和测试代码放到一起的，比如开发的代码很长很长，那么我们如果把开发的代码全部拷贝出来，这样整个文件就显得比较冗余。不方便检查。我们可以使用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)