Wie testen Sie Druckanweisungen?

Question

Sie wollen unittest -Gehäuse für eine Funktion wie das schreiben:

def test_me(a): 
    for b in c: 
     print do_something(a,b) 

Zuerst dachte ich nur die Ausgänge von do_something in einem String zu sammeln und zurückkehr es dann zusammen die gesamte Ausgabe zu drucken und zu testen. Aber es ist nicht immer überzeugend, weil solche Schleifen dazu führen können, dass Ihre Pufferkette sehr groß wird, abhängig von den Umständen. Was können Sie also tun, um die Ausgabe zu testen, wenn sie gedruckt und nicht zurückgegeben wird?

Answer 1

print druckt auf sys.stdout, die Sie Ihrem eigenen Objekt zuweisen können, wenn Sie es wünschen. Das einzige, was Ihr Objekt benötigt, ist eine write-Funktion, die ein einzelnes String-Argument akzeptiert.

Seit Python 2.6 können Sie auch print als Funktion und nicht als Sprachkonstrukt ändern, indem Sie oben in Ihrem Skript from __future__ import print_function hinzufügen. Auf diese Weise können Sie print mit Ihrer eigenen Funktion überschreiben.