def test_resize(self):
buf = SegmentBuilder()
buf.allocate(64)
print()
buf.write_int64(0, 42)
for i in range(63):
buf.allocate(64)
s = buf.as_string()
assert len(s) == 64*64
assert s[:8] == struct.pack('q', 42)
assert s[8:] == b'\x00' * (64*64-8)
def test_alloc_struct(self):
buf = SegmentBuilder(64)
buf.allocate(16)
a = buf.alloc_struct(0, data_size=3, ptrs_size=0)
b = buf.alloc_struct(8, data_size=1, ptrs_size=0)
buf.write_int64(a, 1)
buf.write_int64(a+8, 2)
buf.write_int64(a+16, 3)
buf.write_int64(b, 4)
s = buf.as_string()
assert s == ('\x04\x00\x00\x00\x03\x00\x00\x00' # ptr to a (3, 0)
'\x0c\x00\x00\x00\x01\x00\x00\x00' # ptr to b (1, 0)
'\x01\x00\x00\x00\x00\x00\x00\x00' # a: 1
'\x02\x00\x00\x00\x00\x00\x00\x00' # 2
'\x03\x00\x00\x00\x00\x00\x00\x00' # 3
'\x04\x00\x00\x00\x00\x00\x00\x00') # b: 4
def test_write(self):
expected = struct.pack('<bBhHiIqQfd', 10, 20, 30, 40, 50, 60, 70, 80, 90, 100)
n = len(expected)
buf = SegmentBuilder(n)
buf.allocate(n)
buf.write_int8(0, 10)
buf.write_uint8(1, 20)
buf.write_int16(2, 30)
buf.write_uint16(4, 40)
buf.write_int32(6, 50)
buf.write_uint32(10, 60)
buf.write_int64(14, 70)
buf.write_uint64(22, 80)
buf.write_float32(30, 90)
buf.write_float64(34, 100)
s = buf.as_string()
assert s == expected
def test_write_int64(self):
buf = SegmentBuilder(8)
buf.allocate(8)
buf.write_int64(0, 0x1234ABCD)
s = buf.as_string()
assert s == '\xCD\xAB\x34\x12\x00\x00\x00\x00'