100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
"""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( # pylint: disable=unsupported-binary-operation
|
|
{"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_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"
|