summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_rotor.py
blob: 28c5af5b1c6b0b253325601c9e63a88d51352e61 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import rotor

r = rotor.newrotor("you'll never guess this")
r = rotor.newrotor("you'll never guess this", 12)

A = 'spam and eggs'
B = 'cheese shop'

a = r.encrypt(A)
print `a`
b = r.encryptmore(B)
print `b`

A1 = r.decrypt(a)
print A1
if A1 != A:
    print 'decrypt failed'

B1 = r.decryptmore(b)
print B1
if B1 != B:
    print 'decryptmore failed'

try:
    r.setkey()
except TypeError:
    pass
r.setkey('you guessed it!')