update documentation

test/test_nkode_cipher_keys.py

@@ -0,0 +1,37 @@
+import numpy as np
+import pytest
+from src.models import KeypadSize
+from src.nkode_cipher_v2.nkode_cipher import NKodeCipher
+
+
+@pytest.mark.parametrize(
+    "passcode_len",
+    [
+        6
+    ]
+)
+def test_encode_decode_base64(passcode_len):
+    data = np.random.choice(2**16, passcode_len, replace=False)
+    encoded = NKodeCipher.encode_base64_str(data)
+    decoded = NKodeCipher.decode_base64_str(encoded)
+    assert (len(data) == len(decoded))
+    assert (all(data[idx] == decoded[idx] for idx in range(passcode_len)))
+
+
+@pytest.mark.parametrize(
+    "keypad_size,max_nkode_len",
+    [
+        (KeypadSize(numb_of_keys=10, props_per_key=11), 10),
+        (KeypadSize(numb_of_keys=9, props_per_key=11), 10),
+        (KeypadSize(numb_of_keys=8, props_per_key=11), 12),
+    ])
+def test_decode_mask(keypad_size, max_nkode_len):
+    passcode_entry = np.random.choice(keypad_size.total_props, 4, replace=False)
+    user_keys = NKodeCipher.create(keypad_size, max_nkode_len)
+    passcode_values = [user_keys.property_key[idx] for idx in passcode_entry]
+    enciphered = user_keys.encipher_nkode(passcode_entry)
+    orig_passcode_pos_vals = user_keys.get_props_position_vals(passcode_values)
+    passcode_pos_vals = user_keys.decipher_mask(enciphered.mask, len(passcode_entry))
+    assert (len(passcode_pos_vals) == len(orig_passcode_pos_vals))
+    assert (all(orig_passcode_pos_vals[idx] == passcode_pos_vals[idx] for idx in range(len(passcode_pos_vals))))
+    assert(user_keys.compare_nkode(passcode_entry, enciphered.code))