mirror of
https://github.com/semantica-agi/semantica.git
synced 2026-09-10 04:00:35 +00:00
KaifAhmad1andZohaib Hassan
24e327d161
subprocess.CompletedProcess[str] as a return annotation is not subscriptable at runtime on Python 3.8, causing test collection to abort before any tests run. Adding PEP 563 deferred evaluation makes all annotations strings at import time, restoring 3.8 compatibility without changing behaviour on 3.9+. Co-authored-by: Zohaib Hassan <zohaib179949@gmail.com> Co-authored-by: KaifAhmad1 <kaifahmad087@gmail.com>
121 lines
3.7 KiB
Python
121 lines
3.7 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import textwrap
|
|
from pathlib import Path
|
|
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[2]
|
|
|
|
|
|
def _run_python(code: str) -> subprocess.CompletedProcess[str]:
|
|
env = os.environ.copy()
|
|
env["PYTHONPATH"] = str(REPO_ROOT)
|
|
env["PYTHONDONTWRITEBYTECODE"] = "1"
|
|
return subprocess.run(
|
|
[sys.executable, "-c", textwrap.dedent(code)],
|
|
cwd=REPO_ROOT,
|
|
env=env,
|
|
text=True,
|
|
capture_output=True,
|
|
check=False,
|
|
)
|
|
|
|
|
|
def test_core_extractors_import_from_package() -> None:
|
|
from semantica.semantic_extract import (
|
|
NERExtractor,
|
|
RelationExtractor,
|
|
TripletExtractor,
|
|
)
|
|
|
|
assert NERExtractor.__name__ == "NERExtractor"
|
|
assert RelationExtractor.__name__ == "RelationExtractor"
|
|
assert TripletExtractor.__name__ == "TripletExtractor"
|
|
|
|
|
|
def test_triple_extractor_alias_imports() -> None:
|
|
from semantica.semantic_extract import TripleExtractor, TripletExtractor
|
|
|
|
assert TripleExtractor is TripletExtractor
|
|
|
|
|
|
def test_method_dispatchers_import_without_extractor_cycle() -> None:
|
|
from semantica.semantic_extract.methods import (
|
|
get_entity_method,
|
|
get_relation_method,
|
|
get_triplet_method,
|
|
)
|
|
|
|
assert callable(get_entity_method("pattern"))
|
|
assert callable(get_relation_method("pattern"))
|
|
assert callable(get_triplet_method("pattern"))
|
|
|
|
|
|
def test_import_order_methods_before_extractors() -> None:
|
|
result = _run_python(
|
|
"""
|
|
from semantica.semantic_extract.methods import get_entity_method
|
|
from semantica.semantic_extract import NERExtractor, RelationExtractor, TripletExtractor
|
|
print(get_entity_method("pattern").__name__, NERExtractor.__name__, RelationExtractor.__name__, TripletExtractor.__name__)
|
|
"""
|
|
)
|
|
|
|
assert result.returncode == 0, result.stderr
|
|
assert "extract_entities_pattern NERExtractor RelationExtractor TripletExtractor" in result.stdout
|
|
|
|
|
|
def test_import_order_extractors_before_methods() -> None:
|
|
result = _run_python(
|
|
"""
|
|
from semantica.semantic_extract import NERExtractor, RelationExtractor, TripletExtractor
|
|
from semantica.semantic_extract.methods import get_entity_method
|
|
print(NERExtractor.__name__, RelationExtractor.__name__, TripletExtractor.__name__, get_entity_method("pattern").__name__)
|
|
"""
|
|
)
|
|
|
|
assert result.returncode == 0, result.stderr
|
|
assert "NERExtractor RelationExtractor TripletExtractor extract_entities_pattern" in result.stdout
|
|
|
|
|
|
def test_legacy_type_imports_still_work() -> None:
|
|
from semantica.semantic_extract.ner_extractor import Entity
|
|
from semantica.semantic_extract.relation_extractor import Relation
|
|
from semantica.semantic_extract.triplet_extractor import Triplet
|
|
from semantica.semantic_extract.types import (
|
|
Entity as SharedEntity,
|
|
Relation as SharedRelation,
|
|
Triplet as SharedTriplet,
|
|
)
|
|
|
|
assert Entity is SharedEntity
|
|
assert Relation is SharedRelation
|
|
assert Triplet is SharedTriplet
|
|
|
|
|
|
def test_core_package_imports_do_not_require_yaml() -> None:
|
|
result = _run_python(
|
|
"""
|
|
import importlib.abc
|
|
import sys
|
|
|
|
|
|
class OptionalDependencyBlocker(importlib.abc.MetaPathFinder):
|
|
def find_spec(self, fullname, path=None, target=None):
|
|
if fullname.split(".", 1)[0] == "yaml":
|
|
raise ModuleNotFoundError("No module named 'yaml'")
|
|
return None
|
|
|
|
|
|
sys.meta_path.insert(0, OptionalDependencyBlocker())
|
|
|
|
from semantica.semantic_extract import NERExtractor, RelationExtractor, TripletExtractor
|
|
print(NERExtractor.__name__, RelationExtractor.__name__, TripletExtractor.__name__)
|
|
"""
|
|
)
|
|
|
|
assert result.returncode == 0, result.stderr
|
|
assert "NERExtractor RelationExtractor TripletExtractor" in result.stdout
|