nwb-linkml/nwb_linkml/tests/test_maps.py

43 lines
1.1 KiB
Python
Raw Normal View History

import pytest
import shutil
import tempfile
import yaml
from yaml import CDumper as Dumper
from pathlib import Path
from nwb_linkml.io.schema import load_yaml
def test_preload_maps():
hdmf_style_naming = {
'groups': [
{
'data_type_def': 'Container',
'data_type_inc': 'MainClass',
'doc': 'Demo group',
'datasets': [
{
'data_type_inc': 'Data'
}
]
}
]
}
temp, temp_name = tempfile.mkstemp(suffix='.yaml')
with open(temp_name, 'w') as temp_f:
yaml.dump(hdmf_style_naming, temp_f, Dumper=Dumper)
loaded = load_yaml(Path(temp_name))
assert 'neurodata_type_def' in loaded['groups'][0].keys()
assert 'data_type_def' not in loaded['groups'][0].keys()
assert 'neurodata_type_inc' in loaded['groups'][0].keys()
assert 'data_type_inc' not in loaded['groups'][0].keys()
assert 'neurodata_type_inc' in loaded['groups'][0]['datasets'][0].keys()
assert 'data_type_inc' not in loaded['groups'][0]['datasets'][0].keys()
shutil.rmtree(temp_name)