test_owl2yams.py

import unittest
import pathlib
from rdflib import Graph
from typing import Optional
from owl2yams import owl_model_to_yams
from yams.serialize import serialize_to_python

TEST_DATA = pathlib.Path(__file__).parent / "data"


class Owl2YamsTC(unittest.TestCase):
    def _load_owl_and_yams(
        self,
        owl_model_file_name: str,
        yams_schema_file_name: Optional[str] = None,
        check_cw_uri: bool = False,
    ):
        data_input = TEST_DATA / owl_model_file_name
        owl_model = Graph()
        owl_model.parse(str(data_input), format="turtle")

        schema = owl_model_to_yams(owl_model, "TEST")

        if not check_cw_uri:
            schema.del_entity_type("CWUri")

        if yams_schema_file_name is not None:
            expected_output = TEST_DATA / yams_schema_file_name
            with open(expected_output) as f:
                expected_yams_schema = f.read()
            self.assertEquals(serialize_to_python(schema), expected_yams_schema)
        return schema

    def test_transform_class(self):
        self._load_owl_and_yams("test_class.owl", "test_class.yams")

    def test_transform_attribute(self):
        self._load_owl_and_yams("test_attribute.owl", "test_attribute.yams")

    def test_transform_relation(self):
        self._load_owl_and_yams("test_relation.owl", "test_relation.yams")

    def test_transform_hierarchy(self):
        self._load_owl_and_yams("test_hierarchy.owl", "test_hierarchy.yams")

    def test_transform_relation_without_domain_range(self):
        self._load_owl_and_yams(
            "test_relation_without_domain_range.owl",
            "test_relation_without_domain_range.yams",
        )

    def test_transform_attribute_owl_thing(self):
        self._load_owl_and_yams(
            "test_attribute_owl_thing.owl",
            "test_attribute_owl_thing.yams",
        )

    def test_raise_multi_class(self):
        with self.assertRaises(ValueError):
            self._load_owl_and_yams("test_multi_class.owl")

    def test_raise_multi_attribute(self):
        with self.assertRaises(ValueError):
            self._load_owl_and_yams("test_multi_attribute.owl")

    def test_raise_multi_relation(self):
        with self.assertRaises(ValueError):
            self._load_owl_and_yams("test_multi_relation.owl")

    def test_cwuri(self):
        self._load_owl_and_yams("test_cwuri.owl", "test_cwuri.yams", check_cw_uri=True)


if __name__ == "__main__":
    unittest.main()