def test_fill_from_lock_params(dvc, lock_data):
stage = create_stage(
PipelineStage,
dvc,
PIPELINE_FILE,
deps=["foo"],
outs=["bar"],
params=[
"lorem",
"lorem.ipsum",
{"myparams.yaml": ["ipsum", "foobar"]},
],
)
lock_data["params"] = {
"params.yaml": {
"lorem": "lorem",
"lorem.ipsum": ["i", "p", "s", "u", "m"],
},
"myparams.yaml": {
# missing value in lock for `foobar` params
"ipsum": "ipsum"
},
}
params_deps = split_params_deps(stage)[0]
assert set(params_deps[0].params) == {"lorem", "lorem.ipsum"}
assert set(params_deps[1].params) == {"ipsum", "foobar"}
assert not params_deps[0].hash_info
assert not params_deps[1].hash_info
StageLoader.fill_from_lock(stage, lock_data)
assert params_deps[0].hash_info.value == lock_data["params"]["params.yaml"]
assert (
params_deps[1].hash_info.value == lock_data["params"]["myparams.yaml"]
)
def test_fill_from_lock_missing_params_section(dvc, lock_data):
stage = create_stage(
PipelineStage,
dvc,
PIPELINE_FILE,
deps=["foo"],
outs=["bar"],
params=["lorem", "lorem.ipsum", {"myparams.yaml": ["ipsum"]}],
)
params_deps = split_params_deps(stage)[0]
StageLoader.fill_from_lock(stage, lock_data)
assert not params_deps[0].hash_info and not params_deps[1].hash_info
def test_load_stage_with_params(dvc, stage_data, lock_data):
lock_data["params"] = {"params.yaml": {"lorem": "ipsum"}}
stage_data["params"] = ["lorem"]
dvcfile = Dvcfile(dvc, PIPELINE_FILE)
stage = StageLoader.load_stage(dvcfile, "stage-1", stage_data, lock_data)
params, deps = split_params_deps(stage)
assert deps[0].def_path == "foo" and stage.outs[0].def_path == "bar"
assert params[0].def_path == "params.yaml"
assert params[0].hash_info == HashInfo("params", {"lorem": "ipsum"})
assert deps[0].hash_info == HashInfo("md5", "foo_checksum")
assert stage.outs[0].hash_info == HashInfo("md5", "bar_checksum")