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)