def test_bytearray_to_bucket_num(self):
# valid
bn = BaseDiscretizer.bytearray_to_bucket_num(bytearray([0]))
self.assertIsInstance(bn, int)
self.assertEqual(bn, 0)
bn = BaseDiscretizer.bytearray_to_bucket_num(bytearray([0, 0, 0]))
self.assertEqual(bn, 0)
bn = BaseDiscretizer.bytearray_to_bucket_num(bytearray([3]))
self.assertEqual(bn, 3)
bn = BaseDiscretizer.bytearray_to_bucket_num(bytearray([255]))
self.assertEqual(bn, 255)
bn = BaseDiscretizer.bytearray_to_bucket_num(bytearray([1, 0]))
self.assertEqual(bn, 256)
bn = BaseDiscretizer.bytearray_to_bucket_num(bytearray([1, 1]))
self.assertEqual(bn, 257)
bn = BaseDiscretizer.bytearray_to_bucket_num(bytearray([255, 255]))
self.assertEqual(bn, 65535)
bn = BaseDiscretizer.bytearray_to_bucket_num(bytearray([1, 0, 0]))
self.assertEqual(bn, 65536)
# 0 bytes
self.assertRaises(DiscretizerException,
BaseDiscretizer.bytearray_to_bucket_num,
bytearray())
# not a bytearray
self.assertRaises(DiscretizerException,
BaseDiscretizer.bytearray_to_bucket_num,
0.0)
self.assertRaises(DiscretizerException,
BaseDiscretizer.bytearray_to_bucket_num,
[0, 1, 2])
