def test_setup_config(self, m_merge_configs):
config = {"job_id": 1, "foo": "bar"}
m_merge_configs.return_value = config
result = run.setup_config(["some/config.yaml"])
assert m_merge_configs.called
assert result["job_id"] == "1"
assert result["foo"] == "bar"
def test_setup_config_targets_invalid(self, m_merge_configs):
config = {"targets": range(2), "roles": range(4)}
m_merge_configs.return_value = config
with pytest.raises(AssertionError):
run.setup_config(["some/config.yaml"])
def test_setup_config_targets_ok(self, m_merge_configs):
config = {"targets": range(4), "roles": range(2)}
m_merge_configs.return_value = config
result = run.setup_config(["some/config.yaml"])
assert result["targets"] == [0, 1, 2, 3]
assert result["roles"] == [0, 1]
