42 lines
995 B
Python
42 lines
995 B
Python
import numpy as np
|
|
|
|
from src.keypad import Keypad
|
|
from src.tower_shuffle import TowerShuffle
|
|
|
|
|
|
def test_keypad():
|
|
keypad = Keypad(
|
|
keypad=np.array([
|
|
[8, 9, 10, 11],
|
|
[0, 5, 2, 3],
|
|
[4, 1, 6,7]
|
|
]), k= 3, p=4, keypad_cache=[], tower_shuffler=TowerShuffle.new(3*4))
|
|
|
|
assert keypad.key_entry([8, 5, 6, 11]) == [0,1,2,0]
|
|
|
|
def test_split_shuffle():
|
|
p = 4 # properties_per_key
|
|
k = 3 # number_of_keys
|
|
keypad = Keypad.new_keypad(k, p)
|
|
print(keypad.keypad)
|
|
keypad.split_shuffle()
|
|
print(keypad.keypad)
|
|
|
|
|
|
def test_full_shuffle():
|
|
p = 4 # properties_per_key
|
|
k = 3 # number_of_keys
|
|
keypad = Keypad.new_keypad(k, p)
|
|
print(keypad.keypad)
|
|
keypad.full_shuffle()
|
|
print(keypad.keypad)
|
|
|
|
def test_tower_shuffle():
|
|
p = 4 # properties_per_key
|
|
k = 3 # number_of_keys
|
|
keypad = Keypad.new_keypad(k, p)
|
|
print()
|
|
for _ in range(10):
|
|
print(keypad.keypad)
|
|
keypad.tower_shuffle()
|