def ans_shape()

in benchmarks/db-benchmark/groupby-datafusion.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

def ans_shape(batches):
    rows, cols = 0, 0
    for batch in batches:
        rows += batch.num_rows
        if cols == 0:
            cols = batch.num_columns
        else:
            assert cols == batch.num_columns
    return rows, cols