import doctest
import unittest
import os

flags = (doctest.ELLIPSIS | doctest.NORMALIZE_WHITESPACE |
	 doctest.REPORT_ONLY_FIRST_FAILURE)

files = [os.path.join('doc', filename) for filename in os.listdir('doc')
	 if filename.endswith('.txt')]


def test_suite():
    suite = []
    for testfile in files:
        suite.append(doctest.DocFileTest(testfile, optionflags=flags))
    return unittest.TestSuite(suite)


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