def test_add_sample_int(self):
sum_data = 1
value = 3
sum_aggregation_data = aggregation_data_module.SumAggregationData(
value_type=value_module.ValueLong, sum_data=sum_data)
sum_aggregation_data.add_sample(value, None, None)
self.assertEqual(4, sum_aggregation_data.sum_data)
def test_add_sample_float(self):
sum_data = 1
value = 3.5
sum_aggregation_data = aggregation_data_module.SumAggregationData(
value_type=value_module.ValueDouble, sum_data=sum_data)
sum_aggregation_data.add_sample(value, None, None)
self.assertEqual(4.5, sum_aggregation_data.sum_data)
def test_to_point_int(self):
sum_data = 12
timestamp = datetime(1970, 1, 1)
agg = aggregation_data_module.SumAggregationData(
value_type=value_module.ValueLong, sum_data=sum_data)
converted_point = agg.to_point(timestamp)
self.assertTrue(isinstance(converted_point, point.Point))
self.assertTrue(
isinstance(converted_point.value, value_module.ValueLong))
self.assertEqual(converted_point.value.value, sum_data)
self.assertEqual(converted_point.timestamp, timestamp)
def new_aggregation_data(self, measure):
"""Get a new AggregationData for this aggregation."""
value_type = MetricDescriptorType.to_type_class(
self.get_metric_type(measure))
return aggregation_data.SumAggregationData(value_type=value_type,
sum_data=self._initial_sum)
def test_constructor_int(self):
sum_data = 1
sum_aggregation_data = aggregation_data_module.SumAggregationData(
value_type=value_module.ValueLong, sum_data=sum_data)
self.assertEqual(1, sum_aggregation_data.sum_data)
def test_constructor_float(self):
sum_data = 1.0
sum_aggregation_data = aggregation_data_module.SumAggregationData(
value_type=value_module.ValueDouble, sum_data=sum_data)
self.assertEqual(1.0, sum_aggregation_data.sum_data)
