1818from typing import Sequence
1919from unittest import TestCase
2020
21- from opentelemetry .proto .metrics .v1 .metrics_pb2 import Metric
2221from opentelemetry .sdk .metrics import MeterProvider
23- from opentelemetry .sdk .metrics ._internal .export import MetricExporter , MetricExportResult , PeriodicExportingMetricReader
22+ from opentelemetry .sdk .metrics .export import (
23+ Metric ,
24+ MetricExporter ,
25+ MetricExportResult ,
26+ PeriodicExportingMetricReader ,
27+ )
2428
2529
2630class FakeMetricsExporter (MetricExporter ):
27- def __init__ (self , wait = 0 , preferred_temporality = None , preferred_aggregation = None ):
31+ def __init__ (
32+ self , wait = 0 , preferred_temporality = None , preferred_aggregation = None
33+ ):
2834 self .wait = wait
2935 self .metrics = []
3036 self ._shutdown = False
@@ -34,10 +40,10 @@ def __init__(self, wait=0, preferred_temporality=None, preferred_aggregation=Non
3440 )
3541
3642 def export (
37- self ,
38- metrics_data : Sequence [Metric ],
39- timeout_millis : float = 10_000 ,
40- ** kwargs ,
43+ self ,
44+ metrics_data : Sequence [Metric ],
45+ timeout_millis : float = 10_000 ,
46+ ** kwargs ,
4147 ) -> MetricExportResult :
4248 time .sleep (self .wait )
4349 self .metrics .extend (metrics_data )
@@ -67,7 +73,9 @@ def create_and_shutdown():
6773 return exporter_wr , reader_wr , provider_wr
6874
6975 # When: the provider is shutdown
70- exporter_weakref , reader_weakref , provider_weakref = create_and_shutdown ()
76+ exporter_weakref , reader_weakref , provider_weakref = (
77+ create_and_shutdown ()
78+ )
7179 gc .collect ()
7280
7381 # Then: the provider, exporter and reader should be garbage collected
0 commit comments