diff --git a/tests/test_instrumentation/test_metrics/test_metrics.py b/tests/test_instrumentation/test_metrics/test_metrics.py index a82e7e7b69d924e2e3c98ef20e76fd261f2f4562..b71d828eab8163bf37c7d721006b6afdd13d9fde 100644 --- a/tests/test_instrumentation/test_metrics/test_metrics.py +++ b/tests/test_instrumentation/test_metrics/test_metrics.py @@ -123,6 +123,22 @@ class TestMetric(unittest.TestCase): self.assertTrue(inspect.isgeneratorfunction(generator)) self.assertGreaterEqual(next(iter(metric.to_metric_data().values)).value, 1) + def test_aggregate_metric_measure_generator(self): + my_metric = getMetric('my_metric', 'summary', quantiles=0.5) + metric = AggregateMetric(my_metric) + + @metric.measure() + def generator(): + yield 2 + time.sleep(1) + yield 3 + + for _ in generator(): + pass + self.assertTrue(inspect.isgeneratorfunction(generator)) + self.assertGreaterEqual(next(iter(my_metric.to_metric_data().values)).value, 1) + + def test_quantile_context_manager(self): metric = getMetric('test_metric', 'summary', quantiles=[0.5]) with metric.measure():