staking-deposit-cli/tests/test_utils/test_constants.py
2021-08-23 12:33:04 +02:00

19 lines
535 B
Python

import pytest
from typing import (
Dict,
List,
)
from staking_deposit.utils.constants import _add_index_to_options
@pytest.mark.parametrize(
'arg, test', [
({'en': ['English', 'en']}, {'en': ['1. English', '1', 'English', 'en']}),
({'a': ['a'], 'b': ['b'], 'c': ['c']},
{'a': ['1. a', '1', 'a'], 'b': ['2. b', '2', 'b'], 'c': ['3. c', '3', 'c']})
]
)
def test_add_index_to_options(arg: Dict[str, List[str]], test: Dict[str, List[str]]) -> None:
assert _add_index_to_options(arg) == test