Add t_docstring.py

python/test/t_docstring.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+
+import otmixmod
+import os
+import shutil
+import sys
+from glob import glob
+from tempfile import mkdtemp
+import doctest
+
+mod_path = os.path.dirname(otmixmod.__file__)
+py_files = sorted(glob(mod_path + "/*.py"))
+
+total_failure_count = 0
+total_test_count = 0
+
+# Temporary working directory for IO tests
+work_dir = mkdtemp()
+os.chdir(work_dir)
+
+for py_file in py_files:
+    py_basename = os.path.splitext(os.path.basename(py_file))[0]
+    module = __import__("otmixmod." + py_basename, fromlist=[py_basename])
+    failure_count, test_count = doctest.testmod(
+        module, verbose=False, optionflags=doctest.ELLIPSIS)
+
+    total_failure_count += failure_count
+    total_test_count += test_count
+
+    print(("%s %5d tests failed"
+           % ((py_basename + " ").ljust(60, "."), failure_count)))
+
+print(("-" * 79))
+print(("%s %5d tests failed"
+       % ("TOTAL ".ljust(60, "."), total_failure_count)))
+
+# Delete temporary working directory for IO tests
+os.chdir("..")
+shutil.rmtree(work_dir)
+
+sys.exit(total_failure_count)