| #!/usr/bin/python |
| import wgdata |
| from wvtest import wvtest |
| |
| |
| @wvtest.wvtest |
| def EaterTest(): |
| e = wgdata.Eater('abcdefg') |
| wvtest.WVPASSEQ(e.Eat(2), 'ab') |
| wvtest.WVEXCEPT(wgdata.DecodeError, e.Eat, 10) |
| wvtest.WVPASSEQ(e.Unpack('!3s'), ('cde',)) |
| wvtest.WVPASSEQ(e.Remainder(), 'fg') |
| wvtest.WVPASSEQ(e.Remainder(), '') |
| |
| e = wgdata.Eater('\x01\x02\x03\x04\x05\x06') |
| wvtest.WVPASSEQ(list(e.Iter('!H', 4)), [(0x0102,), (0x0304,)]) |
| wvtest.WVPASSEQ(list(e.Iter('!B', 1)), [(0x05,)]) |
| wvtest.WVEXCEPT(wgdata.DecodeError, lambda: list(e.Iter('!B', 2))) |
| wvtest.WVPASSEQ(e.Remainder(), '\x06') |
| |
| |
| if __name__ == '__main__': |
| wvtest.wvtest_main() |
