summaryrefslogtreecommitdiff
path: root/scripts/update_payload/format_utils_unittest.py
blob: 7153f9edb034f4f8c53cd7c02f566575459094bf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#!/usr/bin/python2
#
# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for format_utils.py."""

import unittest

from update_payload import format_utils


class NumToPercentTest(unittest.TestCase):
  """ Tests number conversion to percentage format."""
  def testHundredPercent(self):
    self.assertEqual(format_utils.NumToPercent(1, 1), '100%')

  def testOverHundredPercent(self):
    self.assertEqual(format_utils.NumToPercent(5, 2), '250%')

  def testWholePercent(self):
    self.assertEqual(format_utils.NumToPercent(3, 10), '30%')

  def testDefaultMinPrecision(self):
    self.assertEqual(format_utils.NumToPercent(3, 9), '33.3%')
    self.assertEqual(format_utils.NumToPercent(3, 900), '0.3%')

  def testDefaultMaxPrecision(self):
    self.assertEqual(format_utils.NumToPercent(3, 9000000), '0.00003%')
    self.assertEqual(format_utils.NumToPercent(3, 90000000), '0%')

  def testCustomMinPrecision(self):
    self.assertEqual(format_utils.NumToPercent(3, 9, min_precision=3),
                     '33.333%')
    self.assertEqual(format_utils.NumToPercent(3, 9, min_precision=0),
                     '33%')

  def testCustomMaxPrecision(self):
    self.assertEqual(format_utils.NumToPercent(3, 900, max_precision=1),
                     '0.3%')
    self.assertEqual(format_utils.NumToPercent(3, 9000, max_precision=1),
                     '0%')


class BytesToHumanReadableTest(unittest.TestCase):
  """ Tests number conversion to human readable format."""
  def testBaseTwo(self):
    self.assertEqual(format_utils.BytesToHumanReadable(0x1000), '4 KiB')
    self.assertEqual(format_utils.BytesToHumanReadable(0x400000), '4 MiB')
    self.assertEqual(format_utils.BytesToHumanReadable(0x100000000), '4 GiB')
    self.assertEqual(format_utils.BytesToHumanReadable(0x40000000000), '4 TiB')

  def testDecimal(self):
    self.assertEqual(format_utils.BytesToHumanReadable(5000, decimal=True),
                     '5 kB')
    self.assertEqual(format_utils.BytesToHumanReadable(5000000, decimal=True),
                     '5 MB')
    self.assertEqual(format_utils.BytesToHumanReadable(5000000000,
                                                       decimal=True),
                     '5 GB')

  def testDefaultPrecision(self):
    self.assertEqual(format_utils.BytesToHumanReadable(5000), '4.8 KiB')
    self.assertEqual(format_utils.BytesToHumanReadable(500000), '488.2 KiB')
    self.assertEqual(format_utils.BytesToHumanReadable(5000000), '4.7 MiB')

  def testCustomPrecision(self):
    self.assertEqual(format_utils.BytesToHumanReadable(5000, precision=3),
                     '4.882 KiB')
    self.assertEqual(format_utils.BytesToHumanReadable(500000, precision=0),
                     '488 KiB')
    self.assertEqual(format_utils.BytesToHumanReadable(5000000, precision=5),
                     '4.76837 MiB')


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