RustPython-RustPython/tests/snippets/bytes_io.py at stack-size-calculation · sthagen/RustPython-RustPython · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

from io import BytesIO

def test_01():
    bytes_string =  b'Test String 1'

    f = BytesIO()
    f.write(bytes_string)

    assert f.getvalue() == bytes_string

def test_02():
    bytes_string =  b'Test String 2'
    f = BytesIO(bytes_string)

    assert f.read() == bytes_string
    assert f.read() == b''

def test_03():
    """
        Tests that the read method (integer arg)
        returns the expected value
    """
    string =  b'Test String 3'
    f = BytesIO(string)

    assert f.read(1) == b'T'
    assert f.read(1) == b'e'
    assert f.read(1) == b's'
    assert f.read(1) == b't'

def test_04():
    """
        Tests that the read method increments the
        cursor position and the seek method moves
        the cursor to the appropriate position
    """
    string =  b'Test String 4'
    f = BytesIO(string)

    assert f.read(4) == b'Test'
    assert f.seek(0) == 0
    assert f.read(4) == b'Test'

if __name__ == "__main__":
    test_01()
    test_02()
    test_03()
    test_04()