mirror of
https://github.com/p2p-ld/nwb-linkml.git
synced 2024-11-10 00:34:29 +00:00
93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
"""
|
|
Placeholder end-to-end tests for generating linkml translations and pydantic models.
|
|
|
|
Should be replaced with more specific unit and integration tests, but in place for now
|
|
to ensure that the basics of the whole thing operate -- not doing any actual data validation
|
|
here.
|
|
"""
|
|
|
|
from pathlib import Path
|
|
from typing import Dict
|
|
|
|
import pytest
|
|
from linkml_runtime.dumpers import yaml_dumper
|
|
from linkml_runtime.linkml_model import SchemaDefinition
|
|
from linkml_runtime.loaders.yaml_loader import YAMLLoader
|
|
|
|
from nwb_linkml.generators.pydantic import NWBPydanticGenerator
|
|
from nwb_linkml.lang_elements import NwbLangSchema
|
|
from nwb_linkml.providers import LinkMLProvider, PydanticProvider
|
|
|
|
|
|
@pytest.mark.dev
|
|
def test_generate_nwblang(tmp_output_dir):
|
|
output_file = (tmp_output_dir / NwbLangSchema.name).with_suffix(".yml")
|
|
yaml_dumper.dump(NwbLangSchema, output_file)
|
|
|
|
|
|
@pytest.mark.dev
|
|
def test_generate_core(nwb_core_fixture, tmp_output_dir):
|
|
schemas = nwb_core_fixture.build().schemas
|
|
|
|
(tmp_output_dir / "schema").mkdir(exist_ok=True)
|
|
|
|
for schema in schemas:
|
|
output_file = tmp_output_dir / "schema" / (schema.name + ".yaml")
|
|
yaml_dumper.dump(schema, output_file)
|
|
|
|
|
|
def load_schema_files(path: Path) -> Dict[str, SchemaDefinition]:
|
|
yaml_loader = YAMLLoader()
|
|
sch: SchemaDefinition
|
|
preloaded_schema = {}
|
|
for schema_path in (path / "schema").glob("*.yaml"):
|
|
sch = yaml_loader.load(str(schema_path), target_class=SchemaDefinition)
|
|
preloaded_schema[sch.name] = sch
|
|
return preloaded_schema
|
|
|
|
|
|
@pytest.mark.dev
|
|
@pytest.mark.depends(on=["test_generate_core"])
|
|
def test_generate_pydantic(tmp_output_dir):
|
|
|
|
(tmp_output_dir / "models").mkdir(exist_ok=True)
|
|
|
|
preloaded_schema = load_schema_files(tmp_output_dir)
|
|
|
|
for schema_path in (tmp_output_dir / "schema").glob("*.yaml"):
|
|
if not schema_path.exists():
|
|
continue
|
|
# python friendly name
|
|
python_name = schema_path.stem.replace(".", "_").replace("-", "_")
|
|
|
|
pydantic_file = (schema_path.parent.parent / "models" / python_name).with_suffix(".py")
|
|
|
|
generator = NWBPydanticGenerator(
|
|
str(schema_path),
|
|
schema_map=preloaded_schema,
|
|
)
|
|
gen_pydantic = generator.serialize()
|
|
|
|
with open(pydantic_file, "w") as pfile:
|
|
pfile.write(gen_pydantic)
|
|
|
|
# make __init__.py
|
|
with open(tmp_output_dir / "models" / "__init__.py", "w") as initfile:
|
|
initfile.write("# Autogenerated module indicator")
|
|
|
|
|
|
@pytest.mark.linkml
|
|
@pytest.mark.provider
|
|
@pytest.mark.dev
|
|
def test_generate_linkml_provider(tmp_output_dir, nwb_core_fixture):
|
|
provider = LinkMLProvider(path=tmp_output_dir, verbose=False)
|
|
|
|
result = provider.build(nwb_core_fixture)
|
|
|
|
|
|
@pytest.mark.pydantic
|
|
@pytest.mark.provider
|
|
@pytest.mark.dev
|
|
def test_generate_pydantic_provider(tmp_output_dir):
|
|
provider = PydanticProvider(path=tmp_output_dir, verbose=False)
|
|
result = provider.build("core")
|