summaryrefslogtreecommitdiff
path: root/scripts/update_payload/histogram_unittest.py
diff options
context:
space:
mode:
authorAlex Deymo <deymo@google.com>2016-06-27 12:14:03 -0700
committerAlex Deymo <deymo@google.com>2016-06-27 12:20:26 -0700
commitce47ca46a3dd74985ac96e95da22ef7a3b742d25 (patch)
treea8ead3ebdbf56da13feeb984991f51cba1f71577 /scripts/update_payload/histogram_unittest.py
parent773f4323a4f6357b7c79debf0f87ffa658fabec4 (diff)
parentc2538fab9a7fc01c0216520874d711c8a9fbd9d3 (diff)
Merge dev-utils update_payload_library and the paycheck.
The update_payload library is a python library for parsing and handling update payload. This library is used by the payload checker python script also included here. The code is merged from this repo: https://chromium.googlesource.com/chromiumos/platform/dev-util/ Bug: 28797993 TEST='import update_payload'
Diffstat (limited to 'scripts/update_payload/histogram_unittest.py')
-rwxr-xr-xscripts/update_payload/histogram_unittest.py60
1 files changed, 60 insertions, 0 deletions
diff --git a/scripts/update_payload/histogram_unittest.py b/scripts/update_payload/histogram_unittest.py
new file mode 100755
index 00000000..421ff203
--- /dev/null
+++ b/scripts/update_payload/histogram_unittest.py
@@ -0,0 +1,60 @@
+#!/usr/bin/python
+#
+# 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 histogram.py."""
+
+import unittest
+
+import format_utils
+import histogram
+
+
+class HistogramTest(unittest.TestCase):
+
+ @staticmethod
+ def AddHumanReadableSize(size):
+ fmt = format_utils.BytesToHumanReadable(size)
+ return '%s (%s)' % (size, fmt) if fmt else str(size)
+
+ def CompareToExpectedDefault(self, actual_str):
+ expected_str = (
+ 'Yes |################ | 5 (83.3%)\n'
+ 'No |### | 1 (16.6%)'
+ )
+ self.assertEqual(actual_str, expected_str)
+
+ def testExampleHistogram(self):
+ self.CompareToExpectedDefault(str(histogram.Histogram(
+ [('Yes', 5), ('No', 1)])))
+
+ def testFromCountDict(self):
+ self.CompareToExpectedDefault(str(histogram.Histogram.FromCountDict(
+ {'Yes': 5, 'No': 1})))
+
+ def testFromKeyList(self):
+ self.CompareToExpectedDefault(str(histogram.Histogram.FromKeyList(
+ ['Yes', 'Yes', 'No', 'Yes', 'Yes', 'Yes'])))
+
+ def testCustomScale(self):
+ expected_str = (
+ 'Yes |#### | 5 (83.3%)\n'
+ 'No | | 1 (16.6%)'
+ )
+ actual_str = str(histogram.Histogram([('Yes', 5), ('No', 1)], scale=5))
+ self.assertEqual(actual_str, expected_str)
+
+ def testCustomFormatter(self):
+ expected_str = (
+ 'Yes |################ | 5000 (4.8 KiB) (83.3%)\n'
+ 'No |### | 1000 (16.6%)'
+ )
+ actual_str = str(histogram.Histogram(
+ [('Yes', 5000), ('No', 1000)], formatter=self.AddHumanReadableSize))
+ self.assertEqual(actual_str, expected_str)
+
+
+if __name__ == '__main__':
+ unittest.main()