Merge pull request #426 from vladistan/convert-test-walker-utils-to-pytest

Convert test walker utils to pytest

Author: kevinschaper

tests/test_utils/test_walker_utils.py

@@ -1,5 +1,4 @@
 import os
-import unittest
 from copy import deepcopy
 
 from jsonasobj2 import as_dict
@@ -33,83 +32,76 @@ def collect(in_obj: YAMLRoot):
     return n_classes
 
 
-class WalkerUtilsTestCase(unittest.TestCase):
+def test_collector():
     """
-    Tests for linkml_runtime.utils.inference_utils
+    Tests using a transformer to collect statistics on an object
     """
+    # use the schema as a data object
+    obj = yaml_loader.load(SCHEMA, target_class=SchemaDefinition)
+    orig = deepcopy(obj)
+    n_classes = 0
+
+    def collect(in_obj: YAMLRoot):
+        nonlocal n_classes
+        if isinstance(in_obj, ClassDefinition):
+            n_classes += 1
+        return in_obj
+
+    obj_tr = traverse_object_tree(obj, collect)
+    assert n_classes > 1
+    # test object is not mutated
+    assert orig == obj
+    # test transformed object is same as input
+    assert obj_tr == obj
+    assert count_classes(obj) == n_classes
+
+
+def test_mutating_transformer():
+    """
+    Tests using transform_object_tree to apply changes
+    """
+    # use the schema as a data object
+    obj = yaml_loader.load(SCHEMA, target_class=SchemaDefinition)
+    orig = deepcopy(obj)
+    n_changes = 0
+
+    def tr(in_obj: YAMLRoot):
+        nonlocal n_changes
+        if isinstance(in_obj, ClassDefinition):
+            in_obj.comments.append(INSERTED_COMMENT)
+            n_changes += 1
+        return in_obj
+
+    obj_tr = traverse_object_tree(obj, tr)
+    assert n_changes > 1
+    assert orig != obj
+    assert obj_tr == obj
+    obj = SchemaDefinition(**as_dict(obj))
+    assert INSERTED_COMMENT in obj.classes["Person"].comments
+
+
+def test_non_mutating_transformer():
+    """
+    Tests using transform_object_tree to apply changes
+    """
+    # use the schema as a data object
+    obj = yaml_loader.load(SCHEMA, target_class=SchemaDefinition)
+    orig = deepcopy(obj)
+    n_changes = 0
+
+    def tr(in_obj: YAMLRoot):
+        nonlocal n_changes
+        if isinstance(in_obj, ClassDefinition):
+            in_obj.comments.append(INSERTED_COMMENT)
+            n_changes += 1
+        return in_obj
 
-    def test_collector(self):
-        """
-        Tests using a transformer to collect statistics on an object
-        """
-        # use the schema as a data object
-        obj = yaml_loader.load(SCHEMA, target_class=SchemaDefinition)
-        orig = deepcopy(obj)
-        n_classes = 0
-
-        def collect(in_obj: YAMLRoot):
-            nonlocal n_classes
-            if isinstance(in_obj, ClassDefinition):
-                n_classes += 1
-            return in_obj
-
-        obj_tr = traverse_object_tree(obj, collect)
-        self.assertGreater(n_classes, 1)
-        # test object is not mutated
-        self.assertEqual(orig, obj)
-        # test transformed object is same as input
-        self.assertEqual(obj_tr, obj)
-        self.assertEqual(count_classes(obj), n_classes)
-
-    def test_mutating_transformer(self):
-        """
-        Tests using transform_object_tree to apply changes
-        """
-        # use the schema as a data object
-        obj = yaml_loader.load(SCHEMA, target_class=SchemaDefinition)
-        orig = deepcopy(obj)
-        n_changes = 0
-
-        def tr(in_obj: YAMLRoot):
-            nonlocal n_changes
-            if isinstance(in_obj, ClassDefinition):
-                in_obj.comments.append(INSERTED_COMMENT)
-                n_changes += 1
-            return in_obj
-
-        obj_tr = traverse_object_tree(obj, tr)
-        self.assertGreater(n_changes, 1)
-        self.assertNotEqual(orig, obj)
-        self.assertEqual(obj_tr, obj)
-        obj = SchemaDefinition(**as_dict(obj))
-        assert INSERTED_COMMENT in obj.classes["Person"].comments
-
-    def test_non_mutating_transformer(self):
-        """
-        Tests using transform_object_tree to apply changes
-        """
-        # use the schema as a data object
-        obj = yaml_loader.load(SCHEMA, target_class=SchemaDefinition)
-        orig = deepcopy(obj)
-        n_changes = 0
-
-        def tr(in_obj: YAMLRoot):
-            nonlocal n_changes
-            if isinstance(in_obj, ClassDefinition):
-                in_obj.comments.append(INSERTED_COMMENT)
-                n_changes += 1
-            return in_obj
-
-        obj_tr = traverse_object_tree(obj, tr, mutate=False)
-        self.assertGreater(n_changes, 1)
-        # check does not mutate
-        self.assertEqual(orig, obj)
-        # check new object is different
-        self.assertNotEqual(obj_tr, orig)
-        self.assertNotEqual(obj_tr, obj)
-        obj_tr = SchemaDefinition(**as_dict(obj_tr))
-        assert INSERTED_COMMENT in obj_tr.classes["Person"].comments
-
-
-if __name__ == "__main__":
-    unittest.main()
+    obj_tr = traverse_object_tree(obj, tr, mutate=False)
+    assert n_changes > 1
+    # check does not mutate
+    assert orig == obj
+    # check new object is different
+    assert obj_tr != orig
+    assert obj_tr != obj
+    obj_tr = SchemaDefinition(**as_dict(obj_tr))
+    assert INSERTED_COMMENT in obj_tr.classes["Person"].comments