def test_translate_to_histogram_with_max_input(self): counter = DataflowDistributionCounter() counter.add_input(INT64_MAX) histogram = Mock(firstBucketOffset=None, bucketCounts=None) counter.translate_to_histogram(histogram) self.assertEquals(histogram.firstBucketOffset, 57) self.assertEquals(histogram.bucketCounts, [1])
def test_translate_to_histogram_with_max_input(self):
counter = DataflowDistributionCounter()
counter.add_input(INT64_MAX)
histogram = Mock(firstBucketOffset=None, bucketCounts=None)
counter.translate_to_histogram(histogram)
self.assertEqual(histogram.firstBucketOffset, 57)
self.assertEqual(histogram.bucketCounts, [1])
def test_add_input(self):
counter = DataflowDistributionCounter()
expected_buckets = [1, 3, 0, 0, 0, 0, 0, 0, 1, 1]
expected_sum = 1510
expected_first_bucket_index = 1
expected_count = 6
expected_min = 1
expected_max = 1000
for element in [1, 500, 2, 3, 1000, 4]:
counter.add_input(element)
histogram = Mock(firstBucketOffset=None, bucketCounts=None)
counter.translate_to_histogram(histogram)
self.assertEquals(counter.sum, expected_sum)
self.assertEquals(counter.count, expected_count)
self.assertEquals(counter.min, expected_min)
self.assertEquals(counter.max, expected_max)
self.assertEquals(histogram.firstBucketOffset, expected_first_bucket_index)
self.assertEquals(histogram.bucketCounts, expected_buckets)
def test_add_input(self):
counter = DataflowDistributionCounter()
expected_buckets = [1, 3, 0, 0, 0, 0, 0, 0, 1, 1]
expected_sum = 1510
expected_first_bucket_index = 1
expected_count = 6
expected_min = 1
expected_max = 1000
for element in [1, 500, 2, 3, 1000, 4]:
counter.add_input(element)
histogram = Mock(firstBucketOffset=None, bucketCounts=None)
counter.translate_to_histogram(histogram)
self.assertEqual(counter.sum, expected_sum)
self.assertEqual(counter.count, expected_count)
self.assertEqual(counter.min, expected_min)
self.assertEqual(counter.max, expected_max)
self.assertEqual(histogram.firstBucketOffset, expected_first_bucket_index)
self.assertEqual(histogram.bucketCounts, expected_buckets)
def test_translate_distribution_using_dataflow_distribution_counter(self):
counter_update = DataflowDistributionCounter()
counter_update.add_input(1)
counter_update.add_input(3)
metric_proto = dataflow.CounterUpdate()
apiclient.translate_distribution(counter_update, metric_proto)
histogram = mock.Mock(firstBucketOffset=None, bucketCounts=None)
counter_update.translate_to_histogram(histogram)
self.assertEqual(metric_proto.distribution.min.lowBits, counter_update.min)
self.assertEqual(metric_proto.distribution.max.lowBits, counter_update.max)
self.assertEqual(metric_proto.distribution.sum.lowBits, counter_update.sum)
self.assertEqual(
metric_proto.distribution.count.lowBits, counter_update.count)
self.assertEqual(
metric_proto.distribution.histogram.bucketCounts,
histogram.bucketCounts)
self.assertEqual(
metric_proto.distribution.histogram.firstBucketOffset,
histogram.firstBucketOffset)
def test_translate_distribution_counter(self):
counter_update = DataflowDistributionCounter()
counter_update.add_input(1)
counter_update.add_input(3)
metric_proto = dataflow.CounterUpdate()
apiclient.translate_distribution(counter_update, metric_proto)
histogram = mock.Mock(firstBucketOffset=None, bucketCounts=None)
counter_update.translate_to_histogram(histogram)
self.assertEqual(metric_proto.distribution.min.lowBits,
counter_update.min)
self.assertEqual(metric_proto.distribution.max.lowBits,
counter_update.max)
self.assertEqual(metric_proto.distribution.sum.lowBits,
counter_update.sum)
self.assertEqual(metric_proto.distribution.count.lowBits,
counter_update.count)
self.assertEqual(metric_proto.distribution.histogram.bucketCounts,
histogram.bucketCounts)
self.assertEqual(metric_proto.distribution.histogram.firstBucketOffset,
histogram.firstBucketOffset)
