def update(self): writer = TerminalWriter(StringIO()) progressbar = ProgressBar([], writer) widget = DataTransferSpeedWidget() widget.init(progressbar, 100) time.sleep(1) progressbar.step += 50 speed = float(widget.update(progressbar, 100)[:-4]) Assert(speed) > 45.0 Assert(speed) < 55.0 time.sleep(2) progressbar.step += 50 speed = float(widget.update(progressbar, 100)[:-4]) Assert(speed) > 20.0 Assert(speed) < 30.0
def init(self): writer = TerminalWriter(StringIO()) progressbar = ProgressBar([], writer) widget = DataTransferSpeedWidget() Assert(widget.init(progressbar, 100)) == '0kb/s'