Text Blocking

gistfile1.py

def blockify(lines):
    assert len({len(line) for line in lines}) in {0, 1}

    return [''.join(section)
            for section in zip(*lines)]

if __name__ == '__main__':
    assert blockify([]) == []
    assert blockify(['AAA', 'BBB', 'CCC']) == ['ABC', 'ABC', 'ABC']
    assert blockify(['AAAAAAAAAAAAA']) == ['A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A']
    assert blockify(['A', 'A', 'A', 'A', 'A']) == ['AAAAA']