mirror of
https://github.com/p2p-ld/nwb-linkml.git
synced 2024-11-14 02:34:28 +00:00
37 lines
1,002 B
Python
37 lines
1,002 B
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)
|