def test_can_iterate_values_in_order():
expected_entries = [
(1, create_test_entry_en('test1')),
(2, create_test_entry_en("test2")),
(3, create_test_entry_en("test3"))]
map = DataMap()
for (id, entry) in expected_entries:
map.add_entry(id, entry)
found = [(id, entry) for (id, entry) in map.items()]
assert found == expected_entries, "Expected map entries to match"
def transform_dmap(dmap: DataMap, obj_schema):
"""Returns a new datamap,
where the items in the original have run through the marshmallow schema."""
results = DataMap()
for entry_id, entry in dmap.items():
data = entry.to_dict()
(converted, errors) = obj_schema.load(data, many=False) # converted
if errors:
raise Exception(str(errors))
results.add_entry(entry_id, converted)
return results
