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