#!/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()