def test_set_func(set_keys, set_val):
"""Setting values."""
flex = FlexDict()
flex.set(set_keys, set_val)
assert flex[set_keys] == set_val
def test_set_func_overwrite_false():
"""Setting values without overwriting existing ones."""
flex = FlexDict(DATA)
res = flex.set('a', 0, overwrite=False)
assert flex['a'] == DATA['a']
assert res is None
def test_set_func_value_error():
"""Invalid argument for set()."""
flex = FlexDict()
with raises(TypeError):
flex.set({'a': 1}, 1)
def test_set_func_increment():
"""Setting values."""
flex = FlexDict()
for i in range(5):
flex.set(['b', 'c', 'd'], i, increment=True)
assert flex['b', 'c', 'd'] == 10