diff --git a/test/utils.py b/test/utils.py index 6c73571..be7696c 100644 --- a/test/utils.py +++ b/test/utils.py @@ -3,6 +3,9 @@ from psh.commands import BaseCommand from io import StringIO class TestFormatter(BaseCommand): + """Formatter useful for tests. Instead of printing to stdout, it + stores any output inside a stringio buffer. This can be retrieved + with the get_data method.""" def __init__(self, *args, **kwargs): super(TestFormatter, self).__init__(*args, **kwargs)