Formatador de texto com coluna fixa

word_wrap.py

import unittest

def word_wrapper(frase, coluna):
    linha = 0
    palavras = []
    frases = []

    for palavra in frase.split():
        tamanho_palavra = len(palavra) + 1

        if linha > 0 and linha + tamanho_palavra > coluna:
            frases.append(palavras)

            linha = 0
            palavras = []

        palavras.append(palavra)
        linha += tamanho_palavra

    frases.append(palavras)

    return '\n'.join([' '.join(f) for f in frases])

class TestWordWrapper(unittest.TestCase):
    def test_a_1(self):
        """Dado frase='a' e coluna=1 retorna 'a'"""
        self.assertEqual(word_wrapper('a', 1), 'a')

    def test_Lorem_ipsum_dolor_8(self):
        r"""Dado frase='Lorem ipsum dolor' e coluna=8 retorna 'Lorem\nipsum\ndolor'"""
        self.assertEqual(repr(word_wrapper('Lorem ipsum dolor', 8)), repr('Lorem\nipsum\ndolor'))

if __name__ == '__main__':
    unittest.main()