def test_set(self):
''' Test that the set bit operation is correct '''
barray = BitArray(array=[0xEC])
for pos in [0, 1, 4]: barray.set(pos)
self.assertEqual([0xFF], barray.to_byte_list())
barray = BitArray(array=[0x00])
for pos in [0, 1, 4]: barray.set(pos)
self.assertEqual([0x13], barray.to_byte_list())
def test_clear(self):
''' Test that the clear bit operation is correct '''
barray = BitArray(array=[0xFF])
for pos in [0, 1, 4]: barray.clear(pos)
self.assertEqual([0xEC], barray.to_byte_list())
barray = BitArray(array=[0x13])
for pos in [0, 1, 4]: barray.clear(pos)
self.assertEqual([0x00], barray.to_byte_list())
def test_flip(self):
''' Test that the flip bit operation is correct '''
barray = BitArray(array=[0xEC])
for pos in [0, 1, 4]: barray.flip(pos)
self.assertEqual([0xFF], barray.to_byte_list())
barray = BitArray(array=[0xFF])
for pos in [0, 1, 4]: barray.flip(pos)
self.assertEqual([0xEC], barray.to_byte_list())
def test_flip_all(self):
''' Test that the flip_all bit operation is correct '''
cases = {
0xEC: [0xFFFFFFFFFFFFFF13],
0xFFFFFFFFFFFFFF13: [0xEC],
}
for array, expected in cases.items():
barray = BitArray(array=[array])
barray.flip_all()
self.assertEqual(expected, barray.to_byte_list())
def test_set_all(self):
''' Test that the set_all bit operation is correct '''
cases = {
0x00: [0xFFFFFFFFFFFFFFFF],
0x12: [0xFFFFFFFFFFFFFFFF],
0x34: [0xFFFFFFFFFFFFFFFF],
}
for array, expected in cases.items():
barray = BitArray(array=[array])
barray.set_all()
self.assertEqual(expected, barray.to_byte_list())
