@@ -321,33 +321,36 @@ def _assert_exemplars_present(
321321 ):
322322 metrics_list = self .memory_metrics_reader .get_metrics_data ()
323323 print (metrics_list )
324- found = { name : 0 for name in metric_names }
324+ metrics = []
325325 for resource_metric in (
326326 getattr (metrics_list , "resource_metrics" , []) or []
327327 ):
328328 for scope_metric in (
329329 getattr (resource_metric , "scope_metrics" , []) or []
330330 ):
331- for metric in getattr (scope_metric , "metrics" , []) or []:
332- if metric .name not in metric_names :
333- continue
334- for point in metric .data .data_points :
335- found [metric .name ] += 1
336- exemplars = getattr (point , "exemplars" , None )
337- self .assertIsNotNone (
338- exemplars ,
339- msg = f"Expected exemplars list attribute on histogram data point for { metric .name } ({ context } )" ,
340- )
341- self .assertGreater (
342- len (exemplars or []),
343- 0 ,
344- msg = f"Expected at least one exemplar on histogram data point for { metric .name } ({ context } ) but none found." ,
345- )
346- for ex in exemplars or []:
347- if hasattr (ex , "span_id" ):
348- self .assertNotEqual (ex .span_id , 0 )
349- if hasattr (ex , "trace_id" ):
350- self .assertNotEqual (ex .trace_id , 0 )
331+ metrics .extend (getattr (scope_metric , "metrics" , []) or [])
332+
333+ found = {name : 0 for name in metric_names }
334+ for metric in metrics :
335+ if metric .name not in metric_names :
336+ continue
337+ for point in metric .data .data_points :
338+ found [metric .name ] += 1
339+ exemplars = getattr (point , "exemplars" , None )
340+ self .assertIsNotNone (
341+ exemplars ,
342+ msg = f"Expected exemplars list attribute on histogram data point for { metric .name } ({ context } )" ,
343+ )
344+ self .assertGreater (
345+ len (exemplars or []),
346+ 0 ,
347+ msg = f"Expected at least one exemplar on histogram data point for { metric .name } ({ context } ) but none found." ,
348+ )
349+ for ex in exemplars or []:
350+ if hasattr (ex , "span_id" ):
351+ self .assertNotEqual (ex .span_id , 0 )
352+ if hasattr (ex , "trace_id" ):
353+ self .assertNotEqual (ex .trace_id , 0 )
351354 for name , count in found .items ():
352355 self .assertGreater (
353356 count ,
0 commit comments