def test_enigmaCase4():
print('\nRunning test case 4')
# Reflector B
# Plugboard Empty
testNigma = crypto.Enigma(rotors={
1: components.Rotor("VI", 0, 12),
2: components.Rotor("VII", 0, 0),
3: components.Rotor("VIII", 0, 0),
},
reflector=components.Reflector("B"),
plugboard=components.Plugboard({}))
test_input = open('tests/testCases4.txt', 'r').read().split('\n\n')
for line in test_input:
input_line, output_line = line.split('\n')
assert testNigma.EDcrypt(input_line) == output_line
print('Done!')
def test_enigmaCase3():
print('\nRunning test case 3')
# Reflector B
# Plugboard Empty
# Slowest rotor grund forced to pass Z
# Rotors with double step
testNigma = crypto.Enigma(rotors={
1: components.Rotor("VI", 0, 0),
2: components.Rotor("VII", 0, 0),
3: components.Rotor("VIII", 0, 0),
},
reflector=components.Reflector("B"),
plugboard=components.Plugboard({}))
test_input = open('tests/testCases3.txt', 'r').read().split('\n\n')
for line in test_input:
input_line, output_line = line.split('\n')
assert testNigma.EDcrypt(input_line) == output_line
print('Done!')