def test_to_list(self):
histogram = mapreduce.Histogram()
histogram.add(0)
histogram.add(1)
histogram.add(31)
histogram.add(60)
histogram.add(61)
histogram.add(123)
self.assertEquals(histogram.to_list(), [2, 1, 0, 1])
histogram = mapreduce.Histogram()
histogram.add(121)
self.assertEquals(histogram.to_list(), [0, 0, 0, 1])
def test_get_bin_number(self):
bucket_size = 30
histogram = mapreduce.Histogram(bucket_size)
self.assertEquals(histogram._get_bin_number(0), 0)
self.assertEquals(histogram._get_bin_number(bucket_size), 0)
self.assertEquals(histogram._get_bin_number(bucket_size + 1), 1)
self.assertEquals(histogram._get_bin_number(bucket_size * 2), 1)
self.assertEquals(histogram._get_bin_number((bucket_size * 2) + 1), 2)
def test_add(self):
histogram = mapreduce.Histogram()
histogram.add(0)
histogram.add(1)
histogram.add(31)
histogram.add(60)
histogram.add(61)
histogram.add(123)
self.assertEquals(histogram._values, {0: 2, 1: 2, 2: 1, 4: 1})
def test_get_bin_number(self):
histogram = mapreduce.Histogram()
self.assertEquals(histogram._get_bin_number(0), 0)
self.assertEquals(
histogram._get_bin_number(mapreduce._BUCKET_SIZE_SECONDS), 0)
self.assertEquals(
histogram._get_bin_number(mapreduce._BUCKET_SIZE_SECONDS + 1), 1)
self.assertEquals(
histogram._get_bin_number(mapreduce._BUCKET_SIZE_SECONDS * 2), 1)
self.assertEquals(
histogram._get_bin_number((mapreduce._BUCKET_SIZE_SECONDS * 2) +
1), 2)
def test_to_list_returns_empty_list(self):
histogram = mapreduce.Histogram()
self.assertEquals(histogram.to_list(), [])
def test_get_bin_number_throws_value_error_for_negative_input(self):
histogram = mapreduce.Histogram()
self.assertRaises(ValueError, histogram._get_bin_number, -1)
