import pytest
from src.user_interface import UserInterface


@pytest.mark.parametrize(
    "numb_of_keys,attrs_per_key",
    [(10, 7,)]
)
def test_attr_set_idx(numb_of_keys, attrs_per_key):
    user_interface = UserInterface.new(attrs_per_key, numb_of_keys)
    for attr_idx in range(70):
        user_interface_idx = user_interface.attr_indices[attr_idx]

        assert (attr_idx % attrs_per_key == user_interface_idx % attrs_per_key)