Python modify_dict Exemples

Exemple #1

def test_str_missing():
    data = {
        'iotile-id': '42',
        'key2': 'value2',
    }
    key = ['NEW']
    value = 'NEWVAL'
    expected = {'iotile-id': '42', 'key2': 'value2', 'NEW': 'NEWVAL'}

    with pytest.raises(KeyError):
        result = modify_dict(data, key, value)

    result = modify_dict(data, key, value, create_if_missing=True)
    assert result == expected

Exemple #2

def test_str():
    data = {'iotile-id': '42', 'key2': 'value2'}
    key = ['iotile-id']
    value = 'NEWVAL'
    expected = {'iotile-id': value, 'key2': 'value2'}

    result = modify_dict(data, key, value)
    assert result == expected

    # Try to modify key that doesn't exist
    key = 'MISSING'
    with pytest.raises(KeyError):
        result = modify_dict(data, key, value)

    # Make sure we're not changing the original
    assert data == {'iotile-id': '42', 'key2': 'value2'}

Exemple #3

def test_keychain_includes_a_not_dict():
    data = {'layer1': {'layer2': {'iotile-id': '42'}}, 'key2': 'NOTADICT'}
    key = ['key2', 'layer2', 'iotile-id']
    value = 'NEWVAL'

    with pytest.raises(ValueError):
        result = modify_dict(data, key, value)  # pylint: disable=unused-variable

Exemple #4

def test_two_deep():
    data = {'layer1': {'layer2': {'iotile-id': '42'}}, 'key2': 'value2'}
    key = ['layer1', 'layer2', 'iotile-id']
    value = 'NEWVAL'
    expected = {'layer1': {'layer2': {'iotile-id': value}}, 'key2': 'value2'}

    result = modify_dict(data, key, value)
    assert result == expected

Exemple #5

def test_one_deep_missing_modify_first_layer():
    data = {'layer1': {'iotile-id': '42'}, 'key2': 'value2'}
    key = ['NEW']
    value = 'NEWVAL'
    expected = {
        'layer1': {
            'iotile-id': '42'
        },
        'key2': 'value2',
        'NEW': 'NEWVAL'
    }

    with pytest.raises(KeyError):
        result = modify_dict(data, key, value)

    result = modify_dict(data, key, value, create_if_missing=True)
    assert result == expected

Exemple #6

def test_one_deep():
    data = {'layer1': {'iotile-id': '42'}, 'key2': 'value2'}
    key = ['layer1', 'iotile-id']
    value = 'NEWVAL'
    expected = {'layer1': {'iotile-id': value}, 'key2': 'value2'}

    result = modify_dict(data, key, value)
    assert result == expected

    # Make sure we're not changing the original
    assert data == {'layer1': {'iotile-id': '42'}, 'key2': 'value2'}