fixing tests

Author: sierra-moxon

linkml_runtime/utils/schemaview.py

@@ -618,6 +618,36 @@ def permissible_value_parent(self, permissible_value: str, enum_name: ENUM_NAME)
         else:
             return []
 
+    @lru_cache()
+    def permissible_value_children(self, permissible_value: str, enum_name: ENUM_NAME) -> Union[
+        str, PermissibleValueText, None, ValueError]:
+        """
+        :param enum_name: parent enum name
+        :param permissible_value: permissible value
+        :return: all direct child enum names (is_a)
+
+        CAT:
+        LION:
+          is_a: CAT
+        ANGRY_LION:
+          is_a: LION
+        BIRD:
+        EAGLE:
+          is_a: BIRD
+
+        """
+
+        enum = self.get_enum(enum_name, strict=True)
+        if enum:
+            if permissible_value in enum.permissible_values:
+                pv = enum.permissible_values[permissible_value]
+                for isapv in enum.permissible_values:
+                    isapv_entity = enum.permissible_values[isapv]
+                    if isapv_entity.is_a and pv.text == isapv_entity.is_a:
+                        return [isapv]
+        else:
+            return []
+
     @lru_cache()
     def slot_parents(self, slot_name: SLOT_NAME, imports=True, mixins=True, is_a=True) -> List[SlotDefinitionName]:
         """
@@ -719,6 +749,22 @@ def permissible_value_ancestors(self, permissible_value_text: str,
                         reflexive=reflexive,
                         depth_first=depth_first)
 
+    @lru_cache()
+    def permissible_value_descendants(self, permissible_value_text: str,
+                                    enum_name: ENUM_NAME,
+                                    reflexive=True,
+                                    depth_first=True) -> List[str]:
+        """
+        Closure of permissible_value_children method
+        :enum
+        """
+
+
+        return _closure(lambda x: self.permissible_value_children(x, enum_name),
+                        permissible_value_text,
+                        reflexive=reflexive,
+                        depth_first=depth_first)
+
     @lru_cache()
     def enum_ancestors(self, enum_name: ENUM_NAME, imports=True, mixins=True, reflexive=True, is_a=True,
                        depth_first=True) -> List[EnumDefinitionName]:

tests/test_utils/test_schemaview.py

@@ -53,9 +53,15 @@ def test_schemaview_enums(self):
                     if pv == "CAT":
                         self.assertEqual(view.permissible_value_parent(pv, e.name), None)
                         self.assertEqual(view.permissible_value_ancestors(pv, e.name), ['CAT'])
+                        self.assertIn("LION", view.permissible_value_descendants(pv, e.name))
+                        self.assertIn("ANGRY_LION", view.permissible_value_descendants(pv, e.name))
+                        self.assertNotIn("EAGLE", view.permissible_value_descendants(pv, e.name))
+                    if pv == "LION":
+                        self.assertIn("ANGRY_LION", view.permissible_value_children(pv, e.name))
                     if pv == "ANGRY_LION":
                         self.assertEqual(view.permissible_value_parent(pv, e.name), ['LION'])
                         self.assertEqual(view.permissible_value_ancestors(pv, e.name), ['ANGRY_LION', 'LION', 'CAT'])
+                        self.assertEquals(["ANGRY_LION"], view.permissible_value_descendants(pv, e.name))
         for cn, c in view.all_classes().items():
             if c.name == "Adult":
                 self.assertEqual(view.class_ancestors(c.name), ['Adult', 'Person', 'HasAliases', 'Thing'])