#!/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 average.py."""

import unittest
from average import Average


class AverageTest(unittest.TestCase):
  """Test average.py."""

  def test_no_record(self) -> None:
    ave = Average("a")
    self.assertEqual(ave.dump().name, "a")
    self.assertEqual(ave.dump().value, 0)

  def test_one_record(self) -> None:
    ave = Average("b")
    ave.record(101)
    self.assertEqual(ave.dump().name, "b")
    self.assertEqual(ave.dump().value, 101)

  def test_more_records(self) -> None:
    ave = Average("c")
    ave.record(1)
    ave.record(2)
    ave.record(3)
    ave.record(4)
    self.assertEqual(ave.dump().name, "c")
    self.assertEqual(ave.dump().value, 2.5)


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