def test_construct_literal_map_from_variable_map():
v = Variable(type=types.LiteralType(simple=types.SimpleType.INTEGER), description="some description")
variable_map = {
'inputa': v,
}
input_txt_dictionary = {'inputa': '15'}
literal_map = helpers.construct_literal_map_from_variable_map(variable_map, input_txt_dictionary)
parsed_literal = literal_map.literals['inputa'].value
ll = literals.Scalar(primitive=literals.Primitive(integer=15))
assert parsed_literal == ll
def test_construct_literal_map_from_parameter_map():
v = Variable(type=types.LiteralType(simple=types.SimpleType.INTEGER), description="some description")
p = Parameter(var=v, required=True)
pm = ParameterMap(parameters={"inputa": p})
input_txt_dictionary = {"inputa": "15"}
literal_map = helpers.construct_literal_map_from_parameter_map(pm, input_txt_dictionary)
parsed_literal = literal_map.literals["inputa"].value
ll = literals.Scalar(primitive=literals.Primitive(integer=15))
assert parsed_literal == ll
with pytest.raises(Exception):
helpers.construct_literal_map_from_parameter_map(pm, {})
def types_to_variable(t: Dict[str, FlyteSdkType]) -> Dict[str, Variable]:
var = {}
if t:
for k, v in t.items():
var[k] = Variable(v.to_flyte_literal_type(), "")
return var