tests for any_of ranges in usage_index()

linkml · May 23, 2024 · ef4f5a5 · ef4f5a5
1 parent 2b90134
commit ef4f5a5
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 1 deletion.
diff --git a/linkml_runtime/utils/schemaview.py b/linkml_runtime/utils/schemaview.py
@@ -1674,6 +1674,7 @@ def usage_index(self) -> Dict[ElementName, List[SchemaUsage]]:
                             if isinstance(x, AnonymousSlotExpression):
                                 x = x.range
                                 k = f"{k}[range]"
+                            k = k.split("[")[0] + "[range]" if "[range]" in k else k
                             u = SchemaUsage(used_by=cn, slot=sn, metaslot=k, used=x)
                             u.inferred = sn in direct_slots
                             ix[x].append(u)

diff --git a/tests/test_utils/input/kitchen_sink_noimports.yaml b/tests/test_utils/input/kitchen_sink_noimports.yaml
@@ -87,9 +87,15 @@ classes:
       - age in years
       - addresses
       - has birth event
+      - reason_for_happiness
     slot_usage:
       name:
         pattern: "^\\S+ \\S+"  ## do not do this in a real schema, people have all kinds of names
+      reason_for_happiness:
+        any_of:
+          - range: BirthEvent
+          - range: EmploymentEvent
+          - range: MarriageEvent
 
   Adult:
     is_a: Person
@@ -353,6 +359,10 @@ slots:
     alias: zip
     range: string
 
+  reason_for_happiness:
+    range: string
+    required: false
+
 
 enums:
   FamilialRelationshipType:

diff --git a/tests/test_utils/test_schemaview.py b/tests/test_utils/test_schemaview.py
@@ -186,7 +186,7 @@ def test_schemaview(self):
 
         self.assertCountEqual(['id', 'name',  ## From Thing
                                'has employment history', 'has familial relationships', 'has medical history',
-                               AGE_IN_YEARS, 'addresses', 'has birth event', ## From Person
+                               AGE_IN_YEARS, 'addresses', 'has birth event', 'reason_for_happiness', ## From Person
                                'aliases'  ## From HasAliases
                                 ],
                               view.class_slots('Person'))
@@ -256,6 +256,42 @@ def test_schemaview(self):
             logging.debug(f' {k} = {v}')
         self.assertIn(SchemaUsage(used_by='FamilialRelationship', slot=RELATED_TO,
                            metaslot='range', used='Person', inferred=False), u['Person'])
+        self.assertListEqual(
+            [SchemaUsage(used_by='Person', 
+                        slot='reason_for_happiness', 
+                        metaslot='any_of[range]', 
+                        used='MarriageEvent', 
+                        inferred=True
+                        ), 
+            SchemaUsage(used_by='Adult',
+                        slot='reason_for_happiness', 
+                        metaslot='any_of[range]', 
+                        used='MarriageEvent', 
+                        inferred=False
+                        )], 
+            u['MarriageEvent'])
+        self.assertListEqual(
+            [SchemaUsage(used_by='Person', 
+                        slot='has employment history', 
+                        metaslot='range', 
+                        used='EmploymentEvent', 
+                        inferred=True), 
+            SchemaUsage(used_by='Person', 
+                        slot='reason_for_happiness', 
+                        metaslot='any_of[range]', 
+                        used='EmploymentEvent', 
+                        inferred=True), 
+            SchemaUsage(used_by='Adult', 
+                        slot='has employment history', 
+                        metaslot='range', 
+                        used='EmploymentEvent', 
+                        inferred=False), 
+            SchemaUsage(used_by='Adult', 
+                        slot='reason_for_happiness', 
+                        metaslot='any_of[range]', 
+                        used='EmploymentEvent', 
+                        inferred=False)],
+            u['EmploymentEvent'])
 
         # test methods also work for attributes
         leaves = view.class_leaves()