def test_udf_data_from_dict_empty():
udf_data = UdfData.from_dict({})
assert udf_data.to_dict() == {
'datacubes': None,
"feature_collection_list": None,
'structured_data_list': None,
'proj': None,
'user_context': {},
}
def test_udf_data_from_dict_structured_data():
udf_data = UdfData.from_dict(
{"structured_data_list": [{
"data": [1, 2, 3]
}]})
assert udf_data.to_dict() == {
'datacubes':
None,
"feature_collection_list":
None,
'structured_data_list': [{
"data": [1, 2, 3],
"type": "list",
"description": "list"
}],
'proj':
None,
'user_context': {},
}
def test_udf_data_from_dict_datacube():
udf_data = UdfData.from_dict(
{"datacubes": [{
"data": [1, 2, 3],
"dimensions": [{
"name": "x"
}]
}]})
assert udf_data.to_dict() == {
'datacubes': [{
"data": [1, 2, 3],
"dimensions": [{
"name": "x"
}]
}],
"feature_collection_list": None,
'structured_data_list': None,
'proj': None,
'user_context': {},
}
def test_udf_data_from_dict_datacube():
udf_data = UdfData.from_dict(
{"datacubes": [{
"data": [1, 2, 3],
"dimensions": [{
"name": "x"
}]
}]})
assert udf_data.to_dict() == {
'datacubes': [{
"data": [1, 2, 3],
"dimensions": [{
"name": "x"
}]
}],
"feature_collection_list": None,
'structured_data_list': None,
'proj': None,
'user_context': {},
}
assert repr(udf_data) \
== "<UdfData datacube_list:[<XarrayDataCube shape:(3,)>] feature_collection_list:[] structured_data_list:[]>"
