@@ -139,8 +139,8 @@ def test_processor_process_exists(self):
139
139
_batch_map [batch_key ] = aggregator
140
140
aggregator2 .update (1.0 )
141
141
processor ._batch_map = _batch_map
142
- record = metrics .Record (metric , labels , aggregator2 )
143
- processor .process (record )
142
+ accumulation = metrics .Accumulation (metric , labels , aggregator2 )
143
+ processor .process (accumulation )
144
144
self .assertEqual (len (processor ._batch_map ), 1 )
145
145
self .assertIsNotNone (processor ._batch_map .get (batch_key ))
146
146
self .assertEqual (processor ._batch_map .get (batch_key ).current , 0 )
@@ -159,8 +159,8 @@ def test_processor_process_not_exists(self):
159
159
batch_key = (metric , SumAggregator , tuple (), labels )
160
160
aggregator .update (1.0 )
161
161
processor ._batch_map = _batch_map
162
- record = metrics .Record (metric , labels , aggregator )
163
- processor .process (record )
162
+ accumulation = metrics .Accumulation (metric , labels , aggregator )
163
+ processor .process (accumulation )
164
164
self .assertEqual (len (processor ._batch_map ), 1 )
165
165
self .assertIsNotNone (processor ._batch_map .get (batch_key ))
166
166
self .assertEqual (processor ._batch_map .get (batch_key ).current , 0 )
@@ -182,8 +182,8 @@ def test_processor_process_not_stateful(self):
182
182
batch_key = (metric , SumAggregator , tuple (), labels )
183
183
aggregator .update (1.0 )
184
184
processor ._batch_map = _batch_map
185
- record = metrics .Record (metric , labels , aggregator )
186
- processor .process (record )
185
+ accumulation = metrics .Accumulation (metric , labels , aggregator )
186
+ processor .process (accumulation )
187
187
self .assertEqual (len (processor ._batch_map ), 1 )
188
188
self .assertIsNotNone (processor ._batch_map .get (batch_key ))
189
189
self .assertEqual (processor ._batch_map .get (batch_key ).current , 0 )
0 commit comments