def test_build_full_config_mini(self) -> None:
expected_mini_config_path = self.input_dir / "expected_mini_config.yml"
expected_mini_config = yaml.load(expected_mini_config_path)
mini_config_path = self.input_dir / "mini_config.yml"
yml_config = yaml.load(mini_config_path)
actual_config = PrivateComputationInfraConfig.build_full_config(
yml_config)
self.assertEqual(actual_config, expected_mini_config)
def gen_config(args: Dict[str, Any]) -> None:
config = yaml.load(args["<input_path>"])
replacements = {}
if "--from" in args and args["--from"] is not None:
other_config = yaml.load(args["--from"])
replacements = build_replacements_from_config(other_config)
update_dict(config, args["--replace"], replacements, args["--accept_all"])
# Coalesce: output to new_output_path if present, otherwise overwrite
output_path = args["<new_output_path>"] or args["<input_path>"]
yaml.dump(config, output_path)
def test_build_full_config_mini_override_error(self) -> None:
mini_override_error_config_path = (self.input_dir /
"mini_override_error_config.yml")
yml_config = yaml.load(mini_override_error_config_path)
with self.assertRaises(CannotFindDependencyError):
PrivateComputationInfraConfig.build_full_config(yml_config)
def from_file(cls, config_file_path: str) -> "ConfigYamlDict":
"""Read a yaml file to a ConfigYamlDict"""
try:
config_dict = yaml.load(Path(config_file_path))
except YAMLError as e:
raise ConfigYamlFileParsingError(config_file_path, str(e))
return ConfigYamlDict.from_dict(config_dict)