qwerrwe / tests /test_dict.py
Nanobit's picture
Ignore unsupported-binary-operation
0dd35c7
raw
history blame
2.96 kB
"""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"