xref: /aosp_15_r20/external/angle/build/util/lib/proto/average_unittests.py (revision 8975f5c5ed3d1c378011245431ada316dfb6f244)
1#!/usr/bin/env vpython3
2
3# Copyright 2024 The Chromium Authors
4# Use of this source code is governed by a BSD-style license that can be
5# found in the LICENSE file.
6"""File for testing average.py."""
7
8import unittest
9from average import Average
10
11
12class AverageTest(unittest.TestCase):
13  """Test average.py."""
14
15  def test_no_record(self) -> None:
16    ave = Average("a")
17    self.assertEqual(ave.dump().name, "a")
18    self.assertEqual(ave.dump().value, 0)
19
20  def test_one_record(self) -> None:
21    ave = Average("b")
22    ave.record(101)
23    self.assertEqual(ave.dump().name, "b")
24    self.assertEqual(ave.dump().value, 101)
25
26  def test_more_records(self) -> None:
27    ave = Average("c")
28    ave.record(1)
29    ave.record(2)
30    ave.record(3)
31    ave.record(4)
32    self.assertEqual(ave.dump().name, "c")
33    self.assertEqual(ave.dump().value, 2.5)
34
35
36if __name__ == '__main__':
37  unittest.main()
38