diff options
| author | Alex Deymo <deymo@google.com> | 2016-06-27 12:14:03 -0700 |
|---|---|---|
| committer | Alex Deymo <deymo@google.com> | 2016-06-27 12:20:26 -0700 |
| commit | ce47ca46a3dd74985ac96e95da22ef7a3b742d25 (patch) | |
| tree | a8ead3ebdbf56da13feeb984991f51cba1f71577 /scripts/update_payload/histogram_unittest.py | |
| parent | 773f4323a4f6357b7c79debf0f87ffa658fabec4 (diff) | |
| parent | c2538fab9a7fc01c0216520874d711c8a9fbd9d3 (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-x | scripts/update_payload/histogram_unittest.py | 60 |
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() |
