blob: 288c82553157a3219cd7bfc9ab01ce88e758cc69 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
# 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.names() == []
assert reg.register("name", "value")
assert reg.names() == ["name"]
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"
assert reg.names() == ["name", "other_name"]
|
