mirror of
https://github.com/p2p-ld/nwb-linkml.git
synced 2024-11-10 00:34:29 +00:00
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
import pdb
|
|
from pathlib import Path
|
|
from typing import Dict
|
|
|
|
import pytest
|
|
import warnings
|
|
|
|
from .fixtures import nwb_core_fixture, tmp_output_dir
|
|
from linkml_runtime.dumpers import yaml_dumper
|
|
from linkml_runtime.linkml_model import SchemaDefinition
|
|
from nwb_linkml.generators.pydantic import NWBPydanticGenerator
|
|
from linkml_runtime.loaders.yaml_loader import YAMLLoader
|
|
|
|
from nwb_linkml.lang_elements import NwbLangSchema
|
|
|
|
def test_generate_nwblang(tmp_output_dir):
|
|
output_file = (tmp_output_dir / NwbLangSchema.name).with_suffix('.yml')
|
|
yaml_dumper.dump(NwbLangSchema, output_file)
|
|
|
|
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.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),
|
|
pydantic_version='2',
|
|
emit_metadata=True,
|
|
gen_classvars=True,
|
|
gen_slots=True,
|
|
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')
|