X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=tests%2Ftest_doctests.py;fp=tests%2Ftest_doctests.py;h=7cd16f74c8ada6c63172776099b229f802305943;hb=27c8005f6dea0026887b80a01b5f93a8f1b3c2b2;hp=0000000000000000000000000000000000000000;hpb=a870050db6bc974b1bb0d132001750b6624fb43f;p=szyfrow.git

diff --git a/tests/test_doctests.py b/tests/test_doctests.py
new file mode 100644
index 0000000..7cd16f7
--- /dev/null
+++ b/tests/test_doctests.py
@@ -0,0 +1,45 @@
+import unittest
+import doctest
+
+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
+
+import support.language_models
+import support.norms
+import support.segment
+import support.text_prettify
+import support.utilities
+
+
+def load_tests(loader, tests, ignore):
+    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')}))
+
+    tests.addTests(doctest.DocTestSuite(support.language_models))
+    tests.addTests(doctest.DocTestSuite(support.norms))
+    tests.addTests(doctest.DocTestSuite(support.segment))
+    tests.addTests(doctest.DocTestSuite(support.text_prettify))
+    tests.addTests(doctest.DocTestSuite(support.utilities))
+
+    return tests