X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=test%2Ftest_doctests.py;h=0af5050de483c8e34b7a274d2743925e62dd4faf;hb=9e70f6e34fa6cb74eb1fe91f39293e6e09e6a77b;hp=31f2a17cdd50627f46484cc9f762d871b6c5619b;hpb=311b300d197536622980f7a837294d8245e326b4;p=cipher-tools.git

diff --git a/test/test_doctests.py b/test/test_doctests.py
index 31f2a17..0af5050 100644
--- a/test/test_doctests.py
+++ b/test/test_doctests.py
@@ -1,7 +1,33 @@
 import unittest
 import doctest
-import cipher
+
+import cipher.caesar
+import cipher.affine
+import cipher.keyword_cipher
+import cipher.polybius
+import cipher.column_transposition
+import cipher.railfence
+import cipher.cadenus
+import cipher.hill
+import cipher.amsco
+import cipher.bifid
+import cipher.autokey
+import cipher.pocket_enigma
+
 
 def load_tests(loader, tests, ignore):
-    tests.addTests(doctest.DocTestSuite(cipher))
+
+    tests.addTests(doctest.DocTestSuite(cipher.caesar))
+    tests.addTests(doctest.DocTestSuite(cipher.affine))
+    tests.addTests(doctest.DocTestSuite(cipher.keyword_cipher))
+    tests.addTests(doctest.DocTestSuite(cipher.polybius))
+    tests.addTests(doctest.DocTestSuite(cipher.column_transposition))
+    tests.addTests(doctest.DocTestSuite(cipher.railfence))
+    tests.addTests(doctest.DocTestSuite(cipher.cadenus))
+    tests.addTests(doctest.DocTestSuite(cipher.hill))
+    tests.addTests(doctest.DocTestSuite(cipher.amsco))
+    tests.addTests(doctest.DocTestSuite(cipher.bifid))
+    tests.addTests(doctest.DocTestSuite(cipher.autokey))
+    tests.addTests(doctest.DocTestSuite(cipher.pocket_enigma, 
+        extraglobs={'pe': cipher.pocket_enigma.PocketEnigma(1, 'a')}))
     return tests