nwb-linkml/nwb_linkml/tests/test_maps/test_maps.py

36 lines
1 KiB
Python

import os
import tempfile
from pathlib import Path
import yaml
from yaml import CDumper as Dumper
from nwb_linkml.io.yaml 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]
assert "data_type_def" not in loaded["groups"][0]
assert "neurodata_type_inc" in loaded["groups"][0]
assert "data_type_inc" not in loaded["groups"][0]
assert "neurodata_type_inc" in loaded["groups"][0]["datasets"][0]
assert "data_type_inc" not in loaded["groups"][0]["datasets"][0]
os.remove(temp_name)