import pytest

from src.basic_data_handling._dynamic_input import ContainsDynamicDict


def test_contains_dynamic_dict_basic_lookup():
    d = ContainsDynamicDict({
        'value': ('x', {'_dynamic': 'number'}),
        'fixed': 'y',
    })

    # direct behavior
    assert 'value' in d
    assert d['value'] == ('x', {'_dynamic': 'number'})

    # dynamic numeric key lookup
    assert 'value1' in d
    assert d['value1'] == ('x', {'_dynamic': 'number'})
    assert 'value999' in d
    assert d['value999'] == ('x', {'_dynamic': 'number'})

    # non-dynamic key and fallback
    assert 'fixed' in d
    assert d['fixed'] == 'y'

    # non-matching key should not be present
    assert 'novalue' not in d
    with pytest.raises(KeyError):
        _ = d['novalue']


def test_contains_dynamic_dict_partial_prefix_not_numeric():
    d = ContainsDynamicDict({'val': ('z', {'_dynamic': 'number'})})

    assert 'val' in d
    assert d['val'] == ('z', {'_dynamic': 'number'})
    assert 'val1' in d
    assert d['val1'] == ('z', {'_dynamic': 'number'})
    assert 'valx' not in d
    with pytest.raises(KeyError):
        _ = d['valx']