mirror of
https://github.com/p2p-ld/nwb-linkml.git
synced 2024-11-10 00:34:29 +00:00
45 lines
1 KiB
Python
45 lines
1 KiB
Python
|
import pytest
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
from nwb_linkml.io.yaml import yaml_peek
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def yaml_file(tmp_path):
|
||
|
|
data = {
|
||
|
|
'key1': 'val1',
|
||
|
|
'key2': 'val2',
|
||
|
|
'key3': {
|
||
|
|
'key1': 'val3',
|
||
|
|
'key4': 'val4'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
out_file = tmp_path / 'test.yaml'
|
||
|
|
with open(out_file, 'w') as yfile:
|
||
|
|
yaml.dump(data, yfile)
|
||
|
|
|
||
|
|
yield out_file
|
||
|
|
|
||
|
|
out_file.unlink()
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
'key,expected,root,first',
|
||
|
|
[
|
||
|
|
('key1', 'val1', True, True),
|
||
|
|
('key1', 'val1', False, True),
|
||
|
|
('key1', ['val1'], True, False),
|
||
|
|
('key1', ['val1', 'val3'], False, False),
|
||
|
|
('key2', 'val2', True, True),
|
||
|
|
('key3', False, True, True),
|
||
|
|
('key4', False, True, True),
|
||
|
|
('key4', 'val4', False, True)
|
||
|
|
]
|
||
|
|
)
|
||
|
|
def test_peek_yaml(key, expected, root, first, yaml_file):
|
||
|
|
if not expected:
|
||
|
|
with pytest.raises(KeyError):
|
||
|
|
_ = yaml_peek(key, yaml_file, root=root, first=first)
|
||
|
|
else:
|
||
|
|
assert yaml_peek(key, yaml_file, root=root, first=first)
|