def test_insert_zeros_2(self):
bv = BitAwareByteArray(bytearray())
bv.zfill(0.5)
bv[0.5:1.5] = BitView([0xFF])
bv.zfill(2.5)
bv[2.5:3.5] = BitView([0])
self.assertEqualBitArrayBitView(self._bitarray_from_bitstring('0000000000000000111111110000'), bv)
def test_add(self):
ba1 = self._create_random_bit_array()
ba2 = self._create_random_bit_array()
ba = ba1 + ba2
bv1 = BitAwareByteArray(self._bitarray_to_bytes(ba1),
stop=float(ba1.length()) / 8)
bv2 = BitAwareByteArray(self._bitarray_to_bytes(ba2),
stop=float(ba2.length()) / 8)
bv = bv1 + bv2
self.assertEqualBitArrayBitView(ba, bv)
def test_setitem__byte(self):
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
buf[0:1] = 3
self.assertEqual([3, 2, 4], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
buf[0.125:1.125] = 3
self.assertEqual([7, 2, 4], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
buf[0.125:1.125] = 0x83
self.assertEqual([7, 3, 4], list(buf))
def test_setitem__bits(self):
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
buf[0:0.125] = 0
self.assertEqual([0, 2, 4], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
buf[0.125:0.25] = 1
self.assertEqual([3, 2, 4], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
buf[1.125:1.375] = 3
self.assertEqual([1, 6, 4], list(buf))
def test_setitem__insert_into_empty_range(self):
buf = BitAwareByteArray(bytearray((1, 2, 4)))
buf[0.125:0.125] = BitView((1, ), 0, 0.125)
self.assertEqual([3, 4, 8, 0], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
buf[0:0] = BitView(bytearray((1, )), 0, 0.125)
self.assertEqual([3, 4, 8, 0], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
buf[0.25:0.25] = BitView(bytearray(1), 0, 0.125)
self.assertEqual([1, 4, 8, 0], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
buf[0.25:0.25] = BitView(bytearray((1, )), 0, 0.125)
self.assertEqual([5, 4, 8, 0], list(buf))
def test_getitem__byte(self):
buf = BitAwareByteArray(bytearray((1, 2, 4)), 0, 3)
self.assertEqual(1, buf[0])
self.assertEqual(2, buf[1])
self.assertEqual(1, buf[1.125])
self.assertEqual(4, buf[2])
self.assertEqual(2, buf[2.125])
self.assertEqual(1, buf[2.25])
self.assertEqual(0, buf[2.375])
buf = BitAwareByteArray(bytearray((128, 1)), 0, 2)
self.assertEqual(3, buf[1 - 0.125])
buf = BitAwareByteArray(bytearray((2, 4)), 0.125, 2)
self.assertEqual(1, buf[0])
self.assertEqual(2, buf[1])
self.assertEqual(1, buf[1.125])
def test_setitem__smaller_val(self):
ba = bitarray('1001010111', endian='little')
bv = BitAwareByteArray(self._bitarray_to_bytes(ba),
stop=float(ba.length()) / 8)
val = bitarray('10', endian='little')
ba[3:7] = val
bv[3.0 / 8:7.0 / 8] = BitView(self._bitarray_to_bytes(val),
stop=2.0 / 8)
self.assertEqualBitArrayBitView(ba, bv)
def test_insert__bytes(self):
buf = BitAwareByteArray(bytearray((1, 2, 4)))
buf.insert(3, bytearray((8, 16)))
self.assertEqual([1, 2, 4, 8, 16], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)))
# 100000000 01000000 00100000
buf.insert(1.25, bytearray((8, 16)))
# 100000000 01 00010000 00001000 000000 00100000
# = 100000000 01000100 00000010 00000000 00100000
# = 1 34 64 0 4
self.assertEqual([1, 34, 64, 0, 4], list(buf))
def test_delitem__bits(self):
buf = BitAwareByteArray(bytearray((1, 2, 4)))
del buf[0:1]
self.assertEqual([2, 4], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)))
del buf[1:]
self.assertEqual([1], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)))
del buf[0:0.125]
self.assertEqual([0, 1, 2], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)))
del buf[0:1.125]
self.assertEqual([1, 2], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)))
del buf[0:2.25]
self.assertEqual([1], list(buf))
def test_getitem__range(self):
buf = BitAwareByteArray(bytearray((1, 2, 4, 128, 1)), 0, 5)
self.assertEqual([1], list(buf[0:1]))
self.assertEqual([1, 2], list(buf[0:2]))
self.assertEqual([2, 4], list(buf[1:3]))
self.assertEqual([2, 4, 128, 1], list(buf[1:]))
self.assertEqual([1, 2], list(buf[:2]))
self.assertEqual([0, 1], list(buf[0.125:2.125]))
self.assertEqual([0, 1], list(buf[0.125:2.125]))
self.assertEqual([128, 1], list(buf[-2:]))
self.assertEqual([1], list(buf[-10:-4]))
self.assertEqual([], list(buf[-10:-5]))
def test_insert_zeros_2(self):
bv = BitAwareByteArray(bytearray())
bv.zfill(0.5)
bv[0.5:1.5] = BitView([0xFF])
bv.zfill(2.5)
bv[2.5:3.5] = BitView([0])
self.assertEqualBitArrayBitView(
self._bitarray_from_bitstring('0000000000000000111111110000'), bv)
def test_iadd_1(self):
a = bytearray(b'\xd3\x94Q`\xb1\x93\x17\xed\xb2W\xa5\x00')
b = bytearray(b'MK\xa3Li\xf9>\x039')
bv1 = BitAwareByteArray(bytearray(a), start=0, stop=11.125)
bv2 = BitView(bytearray(b), start=0, stop=8.75)
bv1 += bv2
a[-1] &= 0x01
a[-1] |= (b[0] & 0x7F) << 1
for i in range(len(b) - 1):
a.append((b[i] >> 7) + ((b[i + 1] & 0x7F) << 1))
self.assertEquals(list(bv1), list(a))
def test_insert__bytes(self):
buf = BitAwareByteArray(bytearray((1, 2, 4)))
buf.insert(3, bytearray((8, 16)))
self.assertEqual([1, 2, 4, 8, 16], list(buf))
buf = BitAwareByteArray(bytearray((1, 2, 4)))
# 100000000 01000000 00100000
buf.insert(1.25, bytearray((8, 16)))
# 100000000 01 00010000 00001000 000000 00100000
# = 100000000 01000100 00000010 00000000 00100000
# = 1 34 64 0 4
self.assertEqual([1, 34, 64, 0, 4], list(buf))
def test_insert_zeros_1(self):
bv = BitAwareByteArray(bytearray((0xFF, 0, 0, 0)))
bv[0:0] = BitView(bytearray((0, )), 0, 0.5)
self.assertEqualBitArrayBitView(
self._bitarray_from_bitstring(
'000000000000000000000000111111110000'), bv)
def test_extend(self):
buf = BitAwareByteArray(bytearray((1, 2, 3)))
buf.extend(bytearray((4, 5)))
self.assertEqual([1, 2, 3, 4, 5], list(buf))
def test_extend(self):
buf = BitAwareByteArray(bytearray((1, 2, 3)))
buf.extend(bytearray((4, 5)))
self.assertEqual([1, 2, 3, 4, 5], list(buf))
def test_insert_zeros(self):
bv = BitAwareByteArray(bytearray(1), 0, 0.5)
bv[0.5:1.5] = BitView((1, ))
self.assertEqualBitArrayBitView(
self._bitarray_from_bitstring('000000010000'), bv)
def test_array_half_byte(self):
a = BitAwareByteArray(bytearray(b'\x02'), start=0, stop=0.5)
self.assertEquals(a[0], 2)
self.assertEquals(list(a), [2])