blob: 95721fcc8886fd294a9544c1e8c22b8dd0b4b46b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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"
|