Test .model.v21.TargetObjectKey.__getitem__

khaeru · Oct 15, 2024 · 6f113ca · 6f113ca
1 parent ae42006
commit 6f113ca
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/sdmx/model/v21.py b/sdmx/model/v21.py
@@ -429,12 +429,17 @@ class TargetIdentifiableObject(TargetObjectValue):
 
 @dataclass
 class TargetObjectKey:
-    """SDMX 2.1 TargetObjectKey."""
+    """SDMX 2.1 TargetObjectKey.
 
+    TargetObjectKey supports item access (:py:`tok["name"]`) to members of
+    :attr:`.key_values`.
+    """
+
+    #: Keys and values of the TargetObjectKey.
     key_values: DictLikeDescriptor[str, TargetObjectValue] = DictLikeDescriptor()
 
     def __getitem__(self, name: str) -> TargetObjectValue:
-        raise NotImplementedError
+        return self.key_values[name]
 
 
 @dataclass

diff --git a/sdmx/tests/model/test_v21.py b/sdmx/tests/model/test_v21.py
@@ -4,6 +4,7 @@
 
 import sdmx
 import sdmx.message
+from sdmx.model import v21
 from sdmx.model import v21 as model
 from sdmx.model.v21 import (
     AttributeDescriptor,
@@ -32,6 +33,7 @@
     MemberSelection,
     MemberValue,
     Observation,
+    TargetObjectKey,
     value_for_dsd_ref,
 )
 
@@ -648,3 +650,15 @@ def test_hierarchy(self, msg: sdmx.message.StructureMessage) -> None:
 
     def test_repr(self, obj: model.HierarchicalCodelist):
         assert "<HierarchicalCodelist HCL_COUNTRY: 1 hierarchies>" == repr(obj)
+
+
+class TestTargetObjectKey:
+    def test_getitem(self) -> None:
+        """:meth:`TargetObjectKey.__getitem__` works."""
+        to = v21.TargetObject(id="FOO")
+        c = Code(id="BAR")
+        tok = TargetObjectKey(
+            key_values={"FOO": v21.TargetIdentifiableObject(value_for=to, obj=c)}
+        )
+
+        assert tok["FOO"].obj is c  # type: ignore [attr-defined]