dkelly/pynkode
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

numpy refactor

Browse Source
This commit is contained in:
Donovan
2025-03-13 04:40:45 -05:00
parent facd9ee318
commit f6bf731186
12 changed files with 261 additions and 140 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test/test_user_cipher_keys.py
View File

@@ -1,8 +1,7 @@
import numpy as np
import pytest
from src.models import KeypadSize
from src.user_cipher import UserCipher, CustomerCipher
from src.utils import generate_random_nonrepeating_list
@pytest.mark.parametrize(
@@ -12,7 +11,8 @@ from src.utils import generate_random_nonrepeating_list
]
)
def test_encode_decode_base64(passcode_len):
data = generate_random_nonrepeating_list(passcode_len)
#data = generate_random_nonrepeating_list(passcode_len)
data = np.random.choice(2**16, passcode_len, replace=False)
encoded = UserCipher.encode_base64_str(data)
decoded = UserCipher.decode_base64_str(encoded)
assert (len(data) == len(decoded))
@@ -28,9 +28,8 @@ def test_encode_decode_base64(passcode_len):
])
def test_decode_mask(keypad_size, max_nkode_len):
customer = CustomerCipher.create(keypad_size)
passcode_entry = generate_random_nonrepeating_list(
keypad_size.numb_of_props,
max_val=keypad_size.numb_of_props)[:4]
#passcode_entry = generate_random_nonrepeating_list(keypad_size.numb_of_props,max_val=keypad_size.numb_of_props)[:4]
passcode_entry = np.random.choice(keypad_size.numb_of_props, 4, replace=False)
passcode_values = [customer.prop_key[idx] for idx in passcode_entry]
set_vals = customer.set_key
user_keys = UserCipher.create(keypad_size, set_vals, max_nkode_len)