diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/test_osmosis.py b/tests/test_osmosis.py
new file mode 100644
index 0000000..82ea31a
--- /dev/null
+++ b/tests/test_osmosis.py
@@ -0,0 +1,40 @@
+from dbt.contracts.graph.manifest import Manifest
+from dbt_osmosis.core.osmosis import DbtYamlManager
+
+
+class TestDbtYamlManager:
+    def test_get_prior_knowledge(test):
+        knowledge = {
+            "myColumn": {
+                "progenitor": "source.my_model.source.Order",
+                "generation": "generation_0",
+                "name": "my_column",
+            },
+            "my_column": {
+                "progenitor": "model.my_model.mart.Order",
+                "generation": "generation_0",
+                "name": "my_column",
+            },
+        }
+        assert (
+            DbtYamlManager.get_prior_knowledge(knowledge, "my_column")["progenitor"]
+            == "source.my_model.source.Order"
+        )
+
+    def test_get_prior_knowledge_with_camel_case(test):
+        knowledge = {
+            "myColumn": {
+                "progenitor": "model.my_model.dwh.Order",
+                "generation": "generation_1",
+                "name": "myColumn",
+            },
+            "my_column": {
+                "progenitor": "model.my_model.mart.Order",
+                "generation": "generation_0",
+                "name": "my_column",
+            },
+        }
+        assert (
+            DbtYamlManager.get_prior_knowledge(knowledge, "my_column")["progenitor"]
+            == "model.my_model.dwh.Order"
+        )