def test_bucket_num_to_bits(self):
# valid
bits = BaseDiscretizer.bucket_num_to_bits(0)
self.assertEqual(bits, '0b00000000')
bits = BaseDiscretizer.bucket_num_to_bits(3)
self.assertEqual(bits, '0b00000011')
bits = BaseDiscretizer.bucket_num_to_bits(255)
self.assertEqual(bits, '0b11111111')
bits = BaseDiscretizer.bucket_num_to_bits(512)
self.assertEqual(bits, '0b'+'00000010'+'00000000')
bits = BaseDiscretizer.bucket_num_to_bits(16744195)
self.assertEqual(bits, '0b'+'11111111'+'01111111'+'00000011')
# not an integer
self.assertRaises(DiscretizerException,
BaseDiscretizer.bucket_num_to_bits,
'abcd')
# bad bucket number
self.assertRaises(DiscretizerException,
BaseDiscretizer.bucket_num_to_bits,
-1)
