def test_seek(): fp = BufferedStream(BytesIO(b"abcdef")) read1 = fp.read(1) assert read1 == b"a" fp.seek(0) read2 = fp.read(1) assert read2 == b"a" read3 = fp.read(2) assert read3 == b"bc" fp.seek(2) read4 = fp.read(2) assert read4 == b"cd" fp.seek(4) read5 = fp.read(2) assert read5 == b"ef"
def test_seek_tell(): fp = BufferedStream(BytesIO(b"abcdef")) read1 = fp.read(1) assert read1 == b"a" assert fp.tell() == 1 fp.seek(0) read2 = fp.read(1) assert read2 == b"a" assert fp.tell() == 1 read3 = fp.read(2) assert read3 == b"bc" assert fp.tell() == 3 fp.seek(2) read4 = fp.read(2) assert read4 == b"cd" assert fp.tell() == 4 fp.seek(4) read5 = fp.read(2) assert read5 == b"ef" assert fp.tell() == 6