diff options
Diffstat (limited to 'tests/test_integers.py')
-rw-r--r-- | tests/test_integers.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/test_integers.py b/tests/test_integers.py index 1e318ab..111f8ab 100644 --- a/tests/test_integers.py +++ b/tests/test_integers.py @@ -2,7 +2,7 @@ import unittest -import rsa +import rsa.core class IntegerTest(unittest.TestCase): @@ -14,10 +14,10 @@ class IntegerTest(unittest.TestCase): message = 42 print "\tMessage: %d" % message - encrypted = rsa.encrypt_int(message, self.pub['e'], self.pub['n']) + encrypted = rsa.core.encrypt_int(message, self.pub.e, self.pub.n) print "\tEncrypted: %d" % encrypted - decrypted = rsa.decrypt_int(encrypted, self.priv['d'], self.pub['n']) + decrypted = rsa.core.decrypt_int(encrypted, self.priv.d, self.pub.n) print "\tDecrypted: %d" % decrypted self.assertEqual(message, decrypted) @@ -26,10 +26,10 @@ class IntegerTest(unittest.TestCase): message = 42 - signed = rsa.encrypt_int(message,self.priv['d'], self.pub['n']) + signed = rsa.core.encrypt_int(message,self.priv.d, self.pub.n) print "\tSigned: %d" % signed - verified = rsa.decrypt_int(signed, self.pub['e'],self.pub['n']) + verified = rsa.core.decrypt_int(signed, self.pub.e,self.pub.n) print "\tVerified: %d" % verified self.assertEqual(message, verified) |