X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=polybius.py;h=79aeb386a5f2920893a83a2f2cef8594a9439080;hb=3d8f7067b9c3a48ef140d7cff834d18ee91f58b3;hp=48509b146ff9b41904aab142de854ae8b6664056;hpb=21c390a77d42729afa23844ef2f1295106bed3de;p=cipher-tools.git

diff --git a/polybius.py b/polybius.py
index 48509b1..79aeb38 100644
--- a/polybius.py
+++ b/polybius.py
@@ -1,6 +1,8 @@
 from utilities import *
 from language_models import *
-from multiprocessing import Pool
+import multiprocessing 
+
+from keyword_cipher import KeywordWrapAlphabet
 
 from logger import logger
 
@@ -142,7 +144,7 @@ def polybius_break_mp(message, column_labels, row_labels,
     """
     if letters_to_merge is None: 
         letters_to_merge = {'j': 'i'}
-    with Pool() as pool:
+    with multiprocessing.Pool() as pool:
         helper_args = [(message, word, wrap, 
                         column_labels, row_labels, column_first, 
                         letters_to_merge, 
@@ -179,3 +181,5 @@ def polybius_break_worker(message, keyword, wrap_alphabet,
                               fit, sanitise(plaintext)[:50]))
     return (keyword, wrap_alphabet, column_order, row_order, column_first), fit
 
+if __name__ == "__main__":
+    import doctest
\ No newline at end of file