import pytest
from src.user_keypad import UserKeypad
from src.models import KeypadSize


@pytest.mark.parametrize(
    "keypad_size",
    [KeypadSize(numb_of_keys=10, props_per_key=11)]
)
def test_prop_set_idx(keypad_size):
    user_keypad = UserKeypad.create(keypad_size)
    for prop_idx in range(keypad_size.numb_of_props):
        user_keypad_idx = user_keypad.keypad[prop_idx]

        assert (prop_idx % keypad_size.props_per_key == user_keypad_idx % keypad_size.props_per_key)