| #!/usr/bin/env vpython3 |
| |
| # Copyright 2024 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| """ A metric implementation to calculate the average of the inputs. """ |
| |
| from measure import Measure |
| from test_script_metrics_pb2 import TestScriptMetric |
| |
| |
| class Average(Measure): |
| |
| def __init__(self, name: str) -> None: |
| self._name = name |
| self._value = 0 |
| self._count = 0 |
| |
| def record(self, value: float) -> None: |
| self._value = (self._value * self._count + value) / (self._count + 1) |
| self._count += 1 |
| |
| def dump(self) -> TestScriptMetric: |
| result = TestScriptMetric() |
| result.name = self._name |
| result.value = self._value |
| return result |
