62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
|
from rdflib import RDF, RDFS, Literal, Namespace, URIRef
|
||
|
|
|
||
|
|
from ontocast.onto.constants import DEFAULT_IRI
|
||
|
|
from ontocast.onto.rdfgraph import RDFGraph
|
||
|
|
from ontocast.tool.agg.normalizer import EntityNormalizer
|
||
|
|
|
||
|
|
|
||
|
|
def test_normalize_string_camel_case(normalizer: EntityNormalizer) -> None:
|
||
|
|
assert normalizer.normalize_string("PLRedShift") == "pl red shift"
|
||
|
|
|
||
|
|
|
||
|
|
def test_normalize_string_snake_case(normalizer: EntityNormalizer) -> None:
|
||
|
|
assert normalizer.normalize_string("PL_red_shift_value") == "pl red shift value"
|
||
|
|
|
||
|
|
|
||
|
|
def test_normalize_string_diacritics(normalizer: EntityNormalizer) -> None:
|
||
|
|
assert normalizer.normalize_string("Café") == "cafe"
|
||
|
|
|
||
|
|
|
||
|
|
def test_normalize_uri_variants(normalizer: EntityNormalizer) -> None:
|
||
|
|
camel_uri = URIRef("http://example.org/PLRedShift")
|
||
|
|
snake_uri = URIRef("http://example.org/PL_red_shift_value")
|
||
|
|
assert normalizer.normalize_uri(camel_uri) == "pl red shift"
|
||
|
|
assert normalizer.normalize_uri(snake_uri) == "pl red shift value"
|
||
|
|
|
||
|
|
|
||
|
|
def test_is_ontology_entity(normalizer: EntityNormalizer) -> None:
|
||
|
|
assert normalizer.is_ontology_entity(URIRef("http://ontology.org/Thing")) is True
|
||
|
|
assert normalizer.is_ontology_entity(URIRef(f"{DEFAULT_IRI}/entity")) is False
|
||
|
|
|
||
|
|
|
||
|
|
def test_create_representation_collects_metadata(normalizer: EntityNormalizer) -> None:
|
||
|
|
graph = RDFGraph()
|
||
|
|
ex = Namespace("http://example.org/")
|
||
|
|
ont = Namespace("http://ontology.org/")
|
||
|
|
|
||
|
|
entity = ex.TestEntity
|
||
|
|
graph.add((entity, RDF.type, ont.Thing))
|
||
|
|
graph.add((entity, RDFS.label, Literal("Test Entity")))
|
||
|
|
graph.add((entity, ex.hasValue, Literal("123")))
|
||
|
|
|
||
|
|
representation = normalizer.create_representation(entity, graph)
|
||
|
|
|
||
|
|
assert representation.entity == entity
|
||
|
|
assert "test entity" in representation.normal_form
|
||
|
|
assert representation.types == [ont.Thing]
|
||
|
|
assert "Test Entity" in representation.labels
|
||
|
|
assert ex.hasValue in representation.properties
|
||
|
|
assert "type" in representation.representation
|
||
|
|
|
||
|
|
|
||
|
|
def test_create_representation_marks_ontology_entity(
|
||
|
|
normalizer: EntityNormalizer,
|
||
|
|
) -> None:
|
||
|
|
graph = RDFGraph()
|
||
|
|
ont = Namespace("http://ontology.org/")
|
||
|
|
entity = ont.SomeClass
|
||
|
|
graph.add((entity, RDF.type, RDFS.Class))
|
||
|
|
|
||
|
|
representation = normalizer.create_representation(entity, graph)
|
||
|
|
assert representation.is_ontology_entity is True
|