Pycrypto (#947)

Dockerfile

@@ -413,6 +413,8 @@ RUN pip install flashtext && \
  pip install pytorch-lightning && \
  pip install datatable && \
  pip install sympy && \
+ # pycrypto is used by competitions team.
+ pip install pycrypto && \
  /tmp/clean-layer.sh
 
 # Tesseract and some associated utility packages

tests/test_pycrypto.py

@@ -0,0 +1,10 @@
+import unittest
+
+from Crypto.Hash import SHA256
+
+class TestPycrypto(unittest.TestCase):
+ def test_digest(self):
+ hash = SHA256.new()
+ hash.update('message'.encode('utf-8'))
+ h = hash.digest()
+ self.assertEqual(h, b'\xabS\n\x13\xe4Y\x14\x98+y\xf9\xb7\xe3\xfb\xa9\x94\xcf\xd1\xf3\xfb"\xf7\x1c\xea\x1a\xfb\xf0+F\x0cm\x1d')