Convert test poly dataclasses to pytest

Author: vladistan

tests/test_utils/test_poly_dataclasses.py

@@ -1,33 +1,26 @@
-import unittest
-
 from linkml_runtime.linkml_model.meta import LINKML, Element
 
 
-class PolyDataclassTestCase(unittest.TestCase):
-    def test_class_for_uri(self):
-        """Test various class lookup options"""
-        e = Element
-
-        # Test class URI
-        cls = e._class_for_uri(LINKML.ClassDefinition)
-        self.assertEqual("ClassDefinition", cls.__name__)
-
-        # Test model URI
-        cls = e._class_for_uri(LINKML.TypeDefinition, use_model_uri=True)
-        self.assertEqual("TypeDefinition", cls.__name__)
+def test_class_for_uri():
+    """Test various class lookup options for polymorphic dataclasses"""
+    e = Element
 
-        # Test class curie (note there isn't any model curie
-        cls = e._class_for_curie("linkml:TypeDefinition")
-        self.assertEqual("TypeDefinition", cls.__name__)
+    # Test class URI lookup
+    cls = e._class_for_uri(LINKML.ClassDefinition)
+    assert cls.__name__ == "ClassDefinition"
 
-        # Make sure the self test works
-        cls = e._class_for_uri(LINKML.Element)
-        self.assertEqual("Element", cls.__name__)
+    # Test model URI lookup
+    cls = e._class_for_uri(LINKML.TypeDefinition, use_model_uri=True)
+    assert cls.__name__ == "TypeDefinition"
 
-        # Make sure we fail gracefully
-        cls = e._class_for_uri("linkml:Missing")
-        self.assertIsNone(cls)
+    # Test class curie lookup (note there isn't any model curie)
+    cls = e._class_for_curie("linkml:TypeDefinition")
+    assert cls.__name__ == "TypeDefinition"
 
+    # Test self lookup works
+    cls = e._class_for_uri(LINKML.Element)
+    assert cls.__name__ == "Element"
 
-if __name__ == "__main__":
-    unittest.main()
+    # Test graceful failure for missing classes
+    cls = e._class_for_uri("linkml:Missing")
+    assert cls is None