def test_zipapply():
    funcs = (add_one, add_one, times_two)
    values = (1, 2, 3)
    result = zipapply(funcs, values)

    assert result == [2, 3, 6]

    result = zipapply([], [])

    assert result == []
Beispiel #2
0
    def get_dest_columns(self, cursor):
        mapping_loaders = map(FunctionMapping.load, self.mapping_signature)

        mappings = zipapply(mapping_loaders, repeat(cursor, len(mapping_loaders)))

        mapping_columns = [m.dest_column for m in mappings]

        return self.transforming_columns() + mapping_columns
Beispiel #3
0
def create_copy_from_line(data_types, row):
    """Return line compatible with COPY FROM command."""
    entity_id, timestamp, attributes = row

    value_mappers = map(value_mapper_by_type.get, data_types)

    values = chain(
        (str(entity_id), str(timestamp)),
        zipapply(value_mappers, attributes)
    )

    return "\t".join(values) + "\n"