|
|
"""Module for testing DictDefault class""" |
|
|
|
|
|
|
|
|
import unittest |
|
|
|
|
|
import pytest |
|
|
|
|
|
from axolotl.utils.dict import DictDefault |
|
|
|
|
|
|
|
|
class DictDefaultTest(unittest.TestCase): |
|
|
""" |
|
|
Test DictDefault class |
|
|
""" |
|
|
|
|
|
def test_dict_default(self): |
|
|
cfg = DictDefault( |
|
|
{ |
|
|
"key_a": {"key_b": "value_a"}, |
|
|
"key_c": "value_c", |
|
|
"key_d": ["value_d", "value_e"], |
|
|
} |
|
|
) |
|
|
|
|
|
assert ( |
|
|
cfg.key_a.key_b == "value_a" |
|
|
), "DictDefault should return value for existing nested keys" |
|
|
|
|
|
assert ( |
|
|
cfg.key_c == "value_c" |
|
|
), "DictDefault should return value for existing keys" |
|
|
|
|
|
assert ( |
|
|
cfg.key_d[0] == "value_d" |
|
|
), "DictDefault should return value for existing keys in list" |
|
|
|
|
|
assert ( |
|
|
"value_e" in cfg.key_d |
|
|
), "DictDefault should support in operator for existing keys in list" |
|
|
|
|
|
def test_dict_or_operator(self): |
|
|
cfg = DictDefault( |
|
|
{ |
|
|
"key_a": {"key_b": "value_a"}, |
|
|
"key_c": "value_c", |
|
|
"key_d": ["value_d", "value_e"], |
|
|
"key_f": "value_f", |
|
|
} |
|
|
) |
|
|
|
|
|
cfg = cfg | DictDefault( |
|
|
{"key_a": {"key_b": "value_b"}, "key_f": "value_g"} |
|
|
) |
|
|
|
|
|
assert ( |
|
|
cfg.key_a.key_b == "value_b" |
|
|
), "DictDefault should support OR operator for existing nested keys" |
|
|
|
|
|
assert cfg.key_c == "value_c", "DictDefault should not delete existing key" |
|
|
|
|
|
assert cfg.key_d == [ |
|
|
"value_d", |
|
|
"value_e", |
|
|
], "DictDefault should not overwrite existing keys in list" |
|
|
|
|
|
assert ( |
|
|
cfg.key_f == "value_g" |
|
|
), "DictDefault should support OR operator for existing key" |
|
|
|
|
|
def test_dict_missingkey(self): |
|
|
cfg = DictDefault({}) |
|
|
|
|
|
assert cfg.random_key is None, "DictDefault should return None for missing keys" |
|
|
|
|
|
def test_dict_or(self): |
|
|
cfg = DictDefault({}) | DictDefault({}) |
|
|
|
|
|
assert ( |
|
|
cfg.random_key is None |
|
|
), "DictDefault should return None for missing keys after | operation" |
|
|
|
|
|
def test_dict_nested_missingparentkey(self): |
|
|
""" |
|
|
Due to subclassing Dict, DictDefault will error if we try to access a nested key whose parent key does not exist. |
|
|
""" |
|
|
cfg = DictDefault({}) |
|
|
|
|
|
with pytest.raises( |
|
|
AttributeError, |
|
|
match=r"'NoneType' object has no attribute 'another_random_key'", |
|
|
): |
|
|
cfg.random_key.another_random_key = "value" |
|
|
|
|
|
def test_dict_shorthand_assignment(self): |
|
|
""" |
|
|
Shorthand assignment is said to not be supported if subclassed. However, their example raises error instead of None. |
|
|
This test ensures that it is supported for current implementation. |
|
|
|
|
|
Ref: https://github.com/mewwts/addict#default-values |
|
|
""" |
|
|
|
|
|
cfg = DictDefault({"key_a": {"key_b": "value_a"}}) |
|
|
|
|
|
cfg.key_a.key_b = "value_b" |
|
|
|
|
|
assert cfg.key_a.key_b == "value_b", "Shorthand assignment should be supported" |
|
|
|
