def write(self): stream = StringIO() writer = TerminalWriter(stream, prefix='spam') writer.indent() progressbar = ProgressBar([], writer) progressbar.write('foo', update=False) Assert(stream.getvalue()) == 'spam foo' progressbar.write('bar') Assert(stream.getvalue()) == 'spam foo\rspam bar'
def get_usable_width(self): writer = TerminalWriter(StringIO(), prefix='spam') Assert(writer.get_usable_width()) == writer.get_width() - len('spam') writer.indent() Assert(writer.get_usable_width()) == writer.get_width() - 8