fix(test): Use proper owl_model_to_yams return types

owl2yams/__init__.py

@@ -3,7 +3,7 @@
 import os
 from pathlib import Path
 import shutil
-from typing import Iterable, Optional, List, Dict
+from typing import Iterable, Optional, List, Dict, Tuple
 
 from argparse import ArgumentParser
 import itertools
@@ -58,7 +58,9 @@
     return fragments
 
 
-def owl_model_to_yams(owl_model: Graph, instance_name: str) -> Schema:
+def owl_model_to_yams(
+    owl_model: Graph, instance_name: str
+) -> Tuple[Schema, Dict[str, str], Dict[str, str]]:
     entitytype_fragment_to_uri: Dict[str, str] = {}
     relationtype_fragment_to_uri: Dict[str, str] = {}
     # 0. Create a new YAMS Schema

test/test_owl2yams.py

@@ -18,7 +18,7 @@
         owl_model = Graph()
         owl_model.parse(str(data_input), format="turtle")
 
-        schema = owl_model_to_yams(owl_model, "TEST")
+        (schema, _, _) = owl_model_to_yams(owl_model, "TEST")
 
         if yams_schema_file_name is not None:
             expected_output = TEST_DATA / yams_schema_file_name