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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
from __future__ import absolute_import
from io import BytesIO
from itertools import combinations_with_replacement
from wvpytest import *
from bup import vint
def encode_and_decode_vuint(x):
f = BytesIO()
vint.write_vuint(f, x)
return vint.read_vuint(BytesIO(f.getvalue()))
def test_vuint():
for x in (0, 1, 42, 128, 10**16, 10**100):
WVPASSEQ(encode_and_decode_vuint(x), x)
WVEXCEPT(Exception, vint.write_vuint, BytesIO(), -1)
WVEXCEPT(EOFError, vint.read_vuint, BytesIO())
def encode_and_decode_vint(x):
f = BytesIO()
vint.write_vint(f, x)
return vint.read_vint(BytesIO(f.getvalue()))
def test_vint():
values = (0, 1, 42, 64, 10**16, 10**100)
for x in values:
WVPASSEQ(encode_and_decode_vint(x), x)
for x in [-x for x in values]:
WVPASSEQ(encode_and_decode_vint(x), x)
WVEXCEPT(EOFError, vint.read_vint, BytesIO())
WVEXCEPT(EOFError, vint.read_vint, BytesIO(b"\x80\x80"))
def encode_and_decode_bvec(x):
f = BytesIO()
vint.write_bvec(f, x)
return vint.read_bvec(BytesIO(f.getvalue()))
def test_bvec():
values = (b'', b'x', b'foo', b'\0', b'\0foo', b'foo\0bar\0')
for x in values:
WVPASSEQ(encode_and_decode_bvec(x), x)
WVEXCEPT(EOFError, vint.read_bvec, BytesIO())
outf = BytesIO()
for x in (b'foo', b'bar', b'baz', b'bax'):
vint.write_bvec(outf, x)
inf = BytesIO(outf.getvalue())
WVPASSEQ(vint.read_bvec(inf), b'foo')
WVPASSEQ(vint.read_bvec(inf), b'bar')
vint.skip_bvec(inf)
WVPASSEQ(vint.read_bvec(inf), b'bax')
def pack_and_unpack(types, *values):
data = vint.pack(types, *values)
return vint.unpack(types, data)
def test_pack_and_unpack():
candidates = (('s', b''),
('s', b'x'),
('s', b'foo'),
('s', b'foo' * 10),
('v', -10**100),
('v', -1),
('v', 0),
('v', 1),
('v', -10**100),
('V', 0),
('V', 1),
('V', 10**100))
WVPASSEQ(pack_and_unpack(''), [])
for f, v in candidates:
WVPASSEQ(pack_and_unpack(f, v), [v])
for (f1, v1), (f2, v2) in combinations_with_replacement(candidates, r=2):
WVPASSEQ(pack_and_unpack(f1 + f2, v1, v2), [v1, v2])
WVEXCEPT(Exception, vint.pack, 's')
WVEXCEPT(Exception, vint.pack, 's', 'foo', 'bar')
WVEXCEPT(Exception, vint.pack, 'x', 1)
WVEXCEPT(Exception, vint.unpack, 's', '')
WVEXCEPT(Exception, vint.unpack, 'x', '')
|
