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
|
from declarativeunittest import *
from construct.lib.binary import *
def test_integer2bits():
assert integer2bits(0, 0) == b""
assert integer2bits(19, 5) == b"\x01\x00\x00\x01\x01"
assert integer2bits(19, 8) == b'\x00\x00\x00\x01\x00\x00\x01\x01'
assert integer2bits(19, 3) == b'\x00\x01\x01'
assert integer2bits(-13, 5) == b"\x01\x00\x00\x01\x01"
assert integer2bits(-13, 8) == b"\x01\x01\x01\x01\x00\x00\x01\x01"
assert raises(integer2bits, 19, -1) == ValueError
assert raises(integer2bits, -19, -1) == ValueError
def test_integer2bytes():
assert integer2bytes(0, 0) == b""
assert integer2bytes(0, 4) == b"\x00\x00\x00\x00"
assert integer2bytes(1, 4) == b"\x00\x00\x00\x01"
assert integer2bytes(255, 4) == b"\x00\x00\x00\xff"
assert integer2bytes(-1, 4) == b"\xff\xff\xff\xff"
assert integer2bytes(-255, 4) == b"\xff\xff\xff\x01"
assert raises(integer2bytes, 19, -1) == ValueError
assert raises(integer2bytes, -19, -1) == ValueError
def test_bits2integer():
assert bits2integer(b"\x01\x00\x00\x01\x01") == 19
assert bits2integer(b"\x01\x00\x00\x01\x01", True) == -13
def test_cross_integers():
for i in [-300,-255,-100,-1,0,1,100,255,300]:
assert bits2integer(integer2bits(i,64),signed=(i<0)) == i
assert bytes2integer(integer2bytes(i,8),signed=(i<0)) == i
assert bits2bytes(integer2bits(i,64)) == integer2bytes(i,8)
assert bytes2bits(integer2bytes(i,8)) == integer2bits(i,64)
def test_bytes2bits():
assert bytes2bits(b"ab") == b"\x00\x01\x01\x00\x00\x00\x00\x01\x00\x01\x01\x00\x00\x00\x01\x00"
assert bytes2bits(b"") == b""
def test_bits2bytes():
assert bits2bytes(b"\x00\x01\x01\x00\x00\x00\x00\x01\x00\x01\x01\x00\x00\x00\x01\x00") == b"ab"
assert bits2bytes(b"") == b""
assert raises(bits2bytes, b"\x00") == ValueError
assert raises(bits2bytes, b"\x00\x00\x00\x00\x00\x00\x00") == ValueError
def test_swapbytes():
assert swapbytes(b"") == b""
assert swapbytes(b"abcd") == b"dcba"
def test_swapbytesinbits():
assert swapbytesinbits(b"0000000011111111") == b"1111111100000000"
assert swapbytesinbits(b"") == b""
assert raises(swapbytesinbits, b"1") == ValueError
def test_swapbitsinbytes():
assert swapbitsinbytes(b'') == b''
assert swapbitsinbytes(b'\xf0') == b'\x0f'
assert swapbitsinbytes(b'\xf0\x00') == b'\x0f\x00'
|
