def test_load_dicts(generator, num_files, expected_size, extensions):
tmp_dir = create_dir(num_files, generator, extensions)
loader = Dicts.from_path(tmp_dir, skip_errors=True)
dicts = list(loader.items)
assert len(dicts) == expected_size, "number of loaded files matches"
for d in dicts:
assert isinstance(d, dict)
assert d, "loaded dict is not empty"
assert "a" in d, "dict loaded correctly"
assert d["a"] < num_files
for f in tmp_dir.iterdir():
f.unlink()
tmp_dir.rmdir()
def load_connection_settings():
""" Retrieve connections from specified yaml."""
directory = Path(os.environ.get(_ENV_VAR_FOR_FILE, _DEFAULT_DIRECTORY))
loader = Dicts.from_path(directory)
return loader