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