diff options
Diffstat (limited to 'tests/test_utils_registry.py')
-rw-r--r-- | tests/test_utils_registry.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tests/test_utils_registry.py b/tests/test_utils_registry.py new file mode 100644 index 0000000..95721fc --- /dev/null +++ b/tests/test_utils_registry.py @@ -0,0 +1,19 @@ +# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-License-Identifier: Apache-2.0 +"""Test the Registry base class.""" +from mlia.utils.registry import Registry + + +def test_registry() -> None: + """Test Registry class.""" + reg = Registry[str]() + assert not str(reg) + assert reg.register("name", "value") + assert not reg.register("name", "value") + assert "name" in reg.items + assert reg.items["name"] == "value" + assert str(reg) + assert reg.register("other_name", "value_2") + assert len(reg.items) == 2 + assert "other_name" in reg.items + assert reg.items["other_name"] == "value_2" |