def test_compress_decompress_field_small(self):
data = np.array([1, 0] * 4, dtype="int32")
fs = agi_bitfield.get_fieldsize(data)
oft = io.BytesIO()
cf = agi_bitfield.compress_field(data, fs, oft)
self.assertEqual(b'', oft.getvalue())
conv_ = agi_bitfield.conv(fs)
dec = agi_bitfield.decode_field(cf[:fs])
self.assertEqual([1 + conv_, 0 + conv_] * 4, dec)
def test_compress_decode_field_overflow(self):
data = np.array([1, 2, 3, 4, 255, 40000, -4, 2], dtype="int32")
fs = agi_bitfield.get_fieldsize(data)
oft = io.BytesIO()
cf = agi_bitfield.compress_field(data, fs, oft)
self.assertEqual(pack("h", 255) + pack("i", 40000), oft.getvalue())
mask_ = agi_bitfield.mask(fs)
dec = agi_bitfield.decode_field(cf)
self.assertEqual([1 + 127, 2 + 127, 3 + 127, 4 + 127, 0xfe & mask_, 0xff & mask_, -4 + 127, 2 + 127], dec)
def test_full_compress_decompress_field(self):
data = np.array([1, 2, 3, 4, 255, -40000, -4, 2], dtype="int32")
fs = agi_bitfield.get_fieldsize(data)
oft = io.BytesIO()
cf = agi_bitfield.compress_field(data, fs, oft)
oft = io.BytesIO(oft.getvalue()) # do this to simulate actual conditions. the data is passed between
# encoding and decoding as bytes so the index would not be preserved
field = agi_bitfield.decode_field(cf[:fs])
agi_bitfield.undo_escapes(field, fs, oft)
self.assertEqual(list(data), field)