1 files changed, 28 insertions, 0 deletions
diff --git a/test/genShake.py b/test/genShake.py
new file mode 100644
index 0000000..ba08981
--- /dev/null
+++ b/test/genShake.py
@@ -0,0 +1,28 @@
+#!/usr/bin/env python2
+import binascii
+
+import sha3
+
+def main():
+    test_vectors = {
+        'Short': 'abc',
+        'Medium': 'abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu',
+        'Long': 'a' * 1000000
+    }
+    
+    hash_funcs = {'SHAKE128': sha3.SHAKE128, 'SHAKE256': sha3.SHAKE256}    
+    output_lens = [31, 63]
+    
+    for hash_name, hash_func in hash_funcs.items():
+        for vector_name, vector_value in test_vectors.items():
+            for output_len in output_lens:
+                print('%s with %s Input and %d bit Output: %s' % (
+                    hash_name,
+                    vector_name,
+                    output_len * 8,
+                    binascii.hexlify(hash_func([ord(c) for c in vector_value], output_len))
+                    )
+                )
+
+if ('__main__' == __name__):
+    main()