#!/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.
"""File for testing metric.py."""

import unittest
from average import Average
from count import Count
from metric import Metric


class MetricTest(unittest.TestCase):
  """Test metric.py."""

  def test_no_metric(self) -> None:
    m = Metric()
    self.assertFalse(m.dump().metrics)

  def test_multiple_metrics(self) -> None:
    m = Metric()
    m.register(Average("a"))
    m.register(Average("b"))
    m.register(Count("c"))
    self.assertEqual(len(m.dump().metrics), 3)


if __name__ == '__main__':
  unittest.main()