mirror of
https://github.com/p2p-ld/nwb-linkml.git
synced 2024-09-19 23:44:28 +00:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import pdb
|
|
|
|
import pytest
|
|
import warnings
|
|
|
|
from .fixtures import nwb_core_fixture, tmp_output_dir
|
|
from linkml_runtime.dumpers import yaml_dumper
|
|
from linkml.generators import PydanticGenerator
|
|
|
|
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
|
|
for schema in schemas:
|
|
output_file = tmp_output_dir / (schema.name + '.yaml')
|
|
yaml_dumper.dump(schema, output_file)
|
|
|
|
@pytest.mark.depends(on=['test_generate_core'])
|
|
def test_generate_pydantic(tmp_output_dir):
|
|
core_file = tmp_output_dir / 'core.yaml'
|
|
pydantic_file = tmp_output_dir / 'core.py'
|
|
|
|
generator = PydanticGenerator(
|
|
str(core_file),
|
|
pydantic_version='1',
|
|
emit_metadata=True,
|
|
gen_classvars=True,
|
|
gen_slots=True
|
|
|
|
)
|
|
gen_pydantic = generator.serialize()
|
|
with open(pydantic_file, 'w') as pfile:
|
|
pfile.write(gen_pydantic)
|