test/test_doctests.py

   1 import unittest
   2 import doctest
   3
   4 import cipher.caesar
   5 import cipher.affine
   6 import cipher.keyword_cipher
   7 import cipher.polybius
   8 import cipher.column_transposition
   9 import cipher.railfence
  10 import cipher.cadenus
  11 import cipher.hill
  12 import cipher.amsco
  13 import cipher.bifid
  14 import cipher.autokey
  15 import cipher.pocket_enigma
  16
  17
  18 def load_tests(loader, tests, ignore):
  19
  20     tests.addTests(doctest.DocTestSuite(cipher.caesar))
  21     tests.addTests(doctest.DocTestSuite(cipher.affine))
  22     tests.addTests(doctest.DocTestSuite(cipher.keyword_cipher))
  23     tests.addTests(doctest.DocTestSuite(cipher.polybius))
  24     tests.addTests(doctest.DocTestSuite(cipher.column_transposition))
  25     tests.addTests(doctest.DocTestSuite(cipher.railfence))
  26     tests.addTests(doctest.DocTestSuite(cipher.cadenus))
  27     tests.addTests(doctest.DocTestSuite(cipher.hill))
  28     tests.addTests(doctest.DocTestSuite(cipher.amsco))
  29     tests.addTests(doctest.DocTestSuite(cipher.bifid))
  30     tests.addTests(doctest.DocTestSuite(cipher.autokey))
  31     tests.addTests(doctest.DocTestSuite(cipher.pocket_enigma,
  32         extraglobs={'pe': cipher.pocket_enigma.PocketEnigma(1, 'a')}))
  33     return tests