1818import sys
1919import unittest
2020
21+
2122class TestMetrics (unittest .TestCase ):
2223 def test_metrics (self ):
2324 """Test that metrics example produces expected values"""
@@ -39,36 +40,64 @@ def test_metrics(self):
3940 output_data = json .loads (result .stdout )
4041
4142 # Get the metrics from the JSON structure
42- metrics = output_data ["resource_metrics" ][0 ]["scope_metrics" ][0 ]["metrics" ]
43+ metrics = output_data ["resource_metrics" ][0 ]["scope_metrics" ][0 ][
44+ "metrics"
45+ ]
4346
4447 # Create a lookup dict for easier testing
4548 metrics_by_name = {metric ["name" ]: metric for metric in metrics }
4649
4750 # Test Counter: should be 1 (called counter.add(1))
48- counter_value = metrics_by_name ["counter" ]["data" ]["data_points" ][0 ]["value" ]
51+ counter_value = metrics_by_name ["counter" ]["data" ]["data_points" ][0 ][
52+ "value"
53+ ]
4954 self .assertEqual (counter_value , 1 , "Counter should have value 1" )
5055
5156 # Test UpDownCounter: should be -4 (1 + (-5) = -4)
52- updown_value = metrics_by_name ["updown_counter" ]["data" ]["data_points" ][0 ]["value" ]
53- self .assertEqual (updown_value , - 4 , "UpDownCounter should have value -4" )
57+ updown_value = metrics_by_name ["updown_counter" ]["data" ][
58+ "data_points"
59+ ][0 ]["value" ]
60+ self .assertEqual (
61+ updown_value , - 4 , "UpDownCounter should have value -4"
62+ )
5463
5564 # Test Histogram: should have count=1, sum=99.9
5665 histogram_data = metrics_by_name ["histogram" ]["data" ]["data_points" ][0 ]
57- self .assertEqual (histogram_data ["count" ], 1 , "Histogram should have count 1" )
58- self .assertEqual (histogram_data ["sum" ], 99.9 , "Histogram should have sum 99.9" )
66+ self .assertEqual (
67+ histogram_data ["count" ], 1 , "Histogram should have count 1"
68+ )
69+ self .assertEqual (
70+ histogram_data ["sum" ], 99.9 , "Histogram should have sum 99.9"
71+ )
5972
6073 # Test Gauge: should be 1 (last value set)
61- gauge_value = metrics_by_name ["gauge" ]["data" ]["data_points" ][0 ]["value" ]
74+ gauge_value = metrics_by_name ["gauge" ]["data" ]["data_points" ][0 ][
75+ "value"
76+ ]
6277 self .assertEqual (gauge_value , 1 , "Gauge should have value 1" )
6378
6479 # Test Observable Counter: should be 1 (from callback)
65- obs_counter_value = metrics_by_name ["observable_counter" ]["data" ]["data_points" ][0 ]["value" ]
66- self .assertEqual (obs_counter_value , 1 , "Observable counter should have value 1" )
80+ obs_counter_value = metrics_by_name ["observable_counter" ]["data" ][
81+ "data_points"
82+ ][0 ]["value" ]
83+ self .assertEqual (
84+ obs_counter_value , 1 , "Observable counter should have value 1"
85+ )
6786
6887 # Test Observable UpDownCounter: should be -10 (from callback)
69- obs_updown_value = metrics_by_name ["observable_updown_counter" ]["data" ]["data_points" ][0 ]["value" ]
70- self .assertEqual (obs_updown_value , - 10 , "Observable updown counter should have value -10" )
88+ obs_updown_value = metrics_by_name ["observable_updown_counter" ][
89+ "data"
90+ ]["data_points" ][0 ]["value" ]
91+ self .assertEqual (
92+ obs_updown_value ,
93+ - 10 ,
94+ "Observable updown counter should have value -10" ,
95+ )
7196
7297 # Test Observable Gauge: should be 9 (from callback)
73- obs_gauge_value = metrics_by_name ["observable_gauge" ]["data" ]["data_points" ][0 ]["value" ]
74- self .assertEqual (obs_gauge_value , 9 , "Observable gauge should have value 9" )
98+ obs_gauge_value = metrics_by_name ["observable_gauge" ]["data" ][
99+ "data_points"
100+ ][0 ]["value" ]
101+ self .assertEqual (
102+ obs_gauge_value , 9 , "Observable gauge should have value 9"
103+ )
0 commit comments