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())