python使用unittest编写测试代码

背景

做开发的朋友在写代码的同时一般都会写测试代码，这对于做运维的同学却很少用。

今天我们就来写写测试代码，用unittest模块。

实例演示

cat test.py

import unittest

def IsOdd(n):

    return n % 2 == 1

class IsOddTests(unittest.TestCase):

    def setUp(self):

        print "Test start..............."

    def tearDown(self):

        print "Test end ............"

    def testOne(self):

        self.failUnless(IsOdd(2))

    def testTwo(self):

        self.failIf(IsOdd(2))

    def testcase1(self):

        self.version = "1.6.1"

        self.assertEqual(self.version, "2.2", "version should be 1.6.1")

if __name__ == '__main__':

    unittest.main()

 

对于unittest一般的用法是把unittest.TestCase作为子类传给一个新建的类，然后在类里面写测试函数。

注意，测试函数名一定要以"test"开头，不然就不会执行的。

"setUp"和"tearDown"是申请和回收资源的函数。