Python TextStream.write_count Examples

Example #1

File: test_Stream.py Project: sonesuke/Kata

def test_load_count():
    stream = io.StringIO()
    a = TextStream(stream)
    a.write_count(3)
    stream.seek(0)
    assert a.load_count() == 3

Example #2

File: test_Stream.py Project: sonesuke/Kata

def test_write_count():
    stream = io.StringIO()
    a = TextStream(stream)
    a.write_count(3)
    stream.seek(0)
    assert stream.readline() == "3\n"