阿小信的博客

doctest其实就是在docstring中按照解释器的格式写出正确运行的效果，实际测试时如果与预计不同就会报错。如下

:::python
#mymodule.py
def split(line, types=None, delimiter=None):
    """
    >>> split('ashin chen xiao baidu')
    ['ashin', 'chen', 'xiao', 'baidu']
    >>> split('ashin 256 512.0', [str, int, float])
    ['ashin', 256, 512.0]
    >>> split('ashin,chen,xiao,baidu', delimiter=',')
    ['ashin', 'chen', 'xiao', 'baidu']
    >>> 
    """

    fields = line.split(delimiter)
    if types:
        fields = [ty(val) for ty, val in zip(types, fields)]
    return fields

def half(x):
    """
    >>> half(6.8)  #改为整数就不会有任何输出
    3.4
    >>> 
    """
    return x/2

if __name__ == '__main__':
    import doctest
    doctest.testmod()

**********************************************************************
File "mymodule.py", line 19, in __main__.half
Failed example:
    half(6.8)
Expected:
    3.4
Got:
    3.3999999999999999
**********************************************************************
1 items had failures:
   1 of   1 in __main__.half
***Test Failed*** 1 failures.

#test.py
import mymodule
import doctest

nfail, ntests = doctest.testmod(mymodule)
print nfail, ntests #0 3

#-*- coding:utf-8 -*-

import mymodule
import unittest

class TestSplitFunc(unittest.TestCase):
    def setUp(self):
        #执行设置操作
        pass
    def tearDown(self):
        #执行清除操作
        pass
    def testsimplestring(self):
        r = mymodule.split('hello world')
        self.assertEqual(r, ['hello', 'world'])
    def testtypeconvert(self):
        r = mymodule.split('ashin 100 200.0', [str, int, float])
        self.assertEqual(r, ['ashin', 100, 200.0])
    def testdelimiter(self):
        r = mymodule.split('a,b,c', delimiter=',')
        self.assertEqual(r, ['a', 'b', 'c'])

#运行unittest
if __name__ == '__main__':
    unittest.main()

...
----------------------------------------------------------------------
Ran 3 tests in 0.001s

OK

F.F
======================================================================
FAIL: testdelimiter (__main__.TestSplitFunc)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "t.py", line 21, in testdelimiter
    self.assertEqual(r, ['a', 'b', 'c'])
AssertionError: ['a,b,c'] != ['a', 'b', 'c']

======================================================================
FAIL: testtypeconvert (__main__.TestSplitFunc)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "t.py", line 18, in testtypeconvert
    self.assertEqual(r, ['ashin', 100, 200.0])
AssertionError: ['ashin', '100', '200.0'] != ['ashin', 100, 200.0]

----------------------------------------------------------------------
Ran 3 tests in 0.001s

FAILED (failures=2)
shell returned 1

import doctest
import unittest
import ttt

suite = unittest.TestSuite()
suite.addTest(doctest.DocTestSuite(ttt))
#or 
#suite = unittest.TestLoader().loadTestsFromTestCase(ttt.ccc) 
runner = unittest.TextTestRunner()
print(runner.run(suite))