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