mirror of
https://github.com/p2p-ld/nwb-linkml.git
synced 2024-11-10 00:34:29 +00:00
sneakers-the-rat
be21325123
- split off generated subclasses into "include" files, not sure if that's good, but in any case we have them separable now. - more work on cleanly un-nesting scalar and 1D-vector data into attributes and lists, respectively - brought the pydantic generator in-repo to do a bunch of overrides
52 lines
1.6 KiB
Python
52 lines
1.6 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.generators.pydantic import NWBPydanticGenerator
|
|
|
|
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)
|
|
|
|
@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'
|
|
|
|
(tmp_output_dir / 'models').mkdir(exist_ok=True)
|
|
|
|
for schema in (tmp_output_dir / 'schema').glob('*.yaml'):
|
|
# python friendly name
|
|
python_name = schema.stem.replace('.', '_').replace('-','_')
|
|
|
|
pydantic_file = (schema.parent.parent / 'models' / python_name).with_suffix('.py')
|
|
|
|
generator = NWBPydanticGenerator(
|
|
str(schema),
|
|
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)
|