python3中对类，对函数进行测试的脚本（单元测试）

先来看一下单元测试的概念：
单元测试：单元测试是用来对一个模块、一个函数或者一个类来进行正确性检验的测试工作。

一、对函数进行测试

  #  要测试的函数：

   import unittest

   def sum(x, y):

         return x + y

    def sub(x, y):

          return x - y

      # 测试该函数的测试类

class mytest(unittest.TestCase):
    # setUp()初始化,
    def setUp(self):
        pass
    # 销毁
    def tearDown(self):
        pass
    # 具体的测试用例，以test开头
    def testsum(self):
        self.assertEqual(sum(1, 2), 3, 'test sum fail')

    def testsub(self):
        self.assertEqual(sub(2, 1), 1, 'test sub fail')

if __name__ == '__main__':
    unittest.main(2)

二、对类进行测试

      #要测试的类

class Person(object):
   def __init__(self, name):
      self.name = name

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

        #对上面的类进行测试（测试脚本）

import unittest
import person


class PersonTest(unittest.TestCase):
   # 初始化
   def setUp(self):
      self.p = person.Person('tom')

   def test_my_add(self):
      inlist = [(1, 2), (-1, -2), (0, 0)]
      qw = [3, -3, 0]
      for i, j in zip(inlist, qw):
         self.assertEqual(self.p.my_add(i[0], i[1]), j, 'test add fail')



if __name__ == '__main__':
   unittest.main()