Added a script test.py working both with Python 2 and 3

1 files changed, 15 insertions, 2 deletions

diff --git a/test.py b/test.py
index 460c36a..ed9873b 100644
--- a/test.py
+++ b/test.py
@@ -3,32 +3,45 @@ Some simple tests
 """
 
 import os
+import sys
+import doctest
 from decorator import decorator
 
+
 @decorator
 def identity(f, *a, **k):
     "do nothing decorator"
     return f(*a, **k)
 
+
 @identity
 def f1():
     "f1"
 
+
 def getfname(func):
     fname = os.path.basename(func.__globals__['__file__'])
     return os.path.splitext(fname)[0] + '.py'
 
+
 def test0():
     this = getfname(identity)
     assert this == 'test.py', this
-    print(identity.__doc__)
+
 
 def test1():
     this = getfname(f1)
     assert this == 'test.py', this
-    print(f1.__doc__)
 
 if __name__ == '__main__':
     for name, test in list(globals().items()):
         if name.startswith('test'):
             test()
+
+    if sys.version >= '3':
+        import documentation3 as doc
+    else:
+        import documentation as doc
+
+    err = doctest.testmod(doc)[0]
+    sys.exit(err)