test.py

import app

def test_generate_text():
    assert len(app.generate_text(10)) == 10
    assert len(app.generate_text(100)) == 100
    assert len(app.generate_text(1000)) == 1000