MLIA-589 Create an API to get target information

Change-Id: Ieeaa9188ea1e29e2ccaad7475d457bce71e3140d

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"