summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_rotor.py
blob: 16a7dcddf69f4efbd9e9c460bc521dd7940a710c (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
29
30
31
32
33
34
import warnings

warnings.filterwarnings("ignore",
                        category=DeprecationWarning,
                        message='.*is deprecated', module=__name__)

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 repr(a)
b = r.encryptmore(B)
print repr(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!')