|
3 | 3 | from .importsupport# flake8: noqa |
4 | 4 | importunittest |
5 | 5 | importcodecs |
6 | | - |
7 | | -fromhtml5lib.inputstreamimportHTMLInputStream,HTMLUnicodeInputStream,HTMLBinaryInputStream |
| 6 | +fromioimportBytesIO |
| 7 | + |
| 8 | +fromhtml5lib.inputstreamimport (BufferedStream,HTMLInputStream, |
| 9 | +HTMLUnicodeInputStream,HTMLBinaryInputStream) |
| 10 | + |
| 11 | +classBufferedStreamTest(unittest.TestCase): |
| 12 | +deftest_basic(self): |
| 13 | +s=b"abc" |
| 14 | +fp=BufferedStream(BytesIO(s)) |
| 15 | +read=fp.read(10) |
| 16 | +assertread==s |
| 17 | + |
| 18 | +deftest_read_length(self): |
| 19 | +fp=BufferedStream(BytesIO(b"abcdef")) |
| 20 | +read1=fp.read(1) |
| 21 | +assertread1==b"a" |
| 22 | +read2=fp.read(2) |
| 23 | +assertread2==b"bc" |
| 24 | +read3=fp.read(3) |
| 25 | +assertread3==b"def" |
| 26 | +read4=fp.read(4) |
| 27 | +assertread4==b"" |
| 28 | + |
| 29 | +deftest_tell(self): |
| 30 | +fp=BufferedStream(BytesIO(b"abcdef")) |
| 31 | +read1=fp.read(1) |
| 32 | +assertfp.tell()==1 |
| 33 | +read2=fp.read(2) |
| 34 | +assertfp.tell()==3 |
| 35 | +read3=fp.read(3) |
| 36 | +assertfp.tell()==6 |
| 37 | +read4=fp.read(4) |
| 38 | +assertfp.tell()==6 |
| 39 | + |
| 40 | +deftest_seek(self): |
| 41 | +fp=BufferedStream(BytesIO(b"abcdef")) |
| 42 | +read1=fp.read(1) |
| 43 | +assertread1==b"a" |
| 44 | +fp.seek(0) |
| 45 | +read2=fp.read(1) |
| 46 | +assertread2==b"a" |
| 47 | +read3=fp.read(2) |
| 48 | +assertread3==b"bc" |
| 49 | +fp.seek(2) |
| 50 | +read4=fp.read(2) |
| 51 | +assertread4==b"cd" |
| 52 | +fp.seek(4) |
| 53 | +read5=fp.read(2) |
| 54 | +assertread5==b"ef" |
| 55 | + |
| 56 | +deftest_seek_tell(self): |
| 57 | +fp=BufferedStream(BytesIO(b"abcdef")) |
| 58 | +read1=fp.read(1) |
| 59 | +assertfp.tell()==1 |
| 60 | +fp.seek(0) |
| 61 | +read2=fp.read(1) |
| 62 | +assertfp.tell()==1 |
| 63 | +read3=fp.read(2) |
| 64 | +assertfp.tell()==3 |
| 65 | +fp.seek(2) |
| 66 | +read4=fp.read(2) |
| 67 | +assertfp.tell()==4 |
| 68 | +fp.seek(4) |
| 69 | +read5=fp.read(2) |
| 70 | +assertfp.tell()==6 |
8 | 71 |
|
9 | 72 |
|
10 | 73 | classHTMLUnicodeInputStreamShortChunk(HTMLUnicodeInputStream): |
|