scurve/test/test_progress.py
2013-12-10 15:12:39 +13:00

32 lines
748 B
Python

import scurve.progress as progress
import StringIO
class TestInplace:
def test_basic(self):
s = StringIO.StringIO()
c = progress.Inplace(stream=s)
assert s.getvalue() == ''
c.tick(10)
assert s.getvalue() == '\r10'
c.tick(10000)
assert s.getvalue() == '\r10\r10000'
c.inject("foo")
c.clear()
def test_nostream(self):
c = progress.Inplace(stream=None)
c.tick(10)
c.clear()
class TestProgress:
def test_basic(self):
s = StringIO.StringIO()
p = progress.Progress(100, stream=s)
p.tick(25)
assert p.prev == 0.25
p.tick(50)
assert p.prev == 0.5
p.full()
assert p.prev == 1.0