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)