def test_overloaded_ctor(self):
b = Bitmap(0x0FF0F0)
self.assertTrue(b)
bitset = b.get_bitmap_as_indices()
self.assertTrue(len(bitset) == 12)
i = 0
for j in range(4, 8):
self.assertEqual(bitset[i], j)
i += 1
for j in range(12, 20):
self.assertEqual(bitset[i], j)
i += 1
def test_is_bit_set(self):
b = Bitmap(0xAAAAAAAA55555555)
bitset = b.get_bitmap_as_indices()
for i in range(64):
if (i < 32):
if (i % 2):
self.assertFalse(b.is_bit_set(i))
else:
self.assertTrue(b.is_bit_set(i))
else:
if (i % 2):
self.assertTrue(b.is_bit_set(i))
else:
self.assertFalse(b.is_bit_set(i))
self.assertRaises(TooDamnMuchDammitError, b.is_bit_set, 1024)
def test_ctor(self):
b = Bitmap()
self.assertTrue(b)
bitset = b.get_bitmap_as_indices()
self.assertTrue(not len(bitset))
def test_max_bits(self):
b = Bitmap(2 ** 1024 - 1)
self.assertTrue(b)
bitset = b.get_bitmap_as_indices()
self.assertTrue(len(bitset) == 1024)
