From 7750d910bb7047c0c24f2e93dd302a445b5068ec Mon Sep 17 00:00:00 2001 From: Sahina Bose Date: Wed, 19 Mar 2014 19:25:16 +0530 Subject: utils: Added utils method to convert size The method converts a value from a given unit to another. Example usage: convertSize (value, 'KB', 'MB) Change-Id: If8e004076493006234be81e656eabde7d3bf9a6a Signed-off-by: Sahina Bose Reviewed-on: https://cuckoo.blr.redhat.com:8443/32 Reviewed-by: Shubhendu Tripathi Reviewed-by: Nishanth Thomas --- tests/utilsTests.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'tests/utilsTests.py') diff --git a/tests/utilsTests.py b/tests/utilsTests.py index 69da659..fdf1ec3 100644 --- a/tests/utilsTests.py +++ b/tests/utilsTests.py @@ -146,3 +146,21 @@ class xml2dictTests(TestCaseBase): utils.xml2dict("not an etree object") self.assertRaises(AttributeError, _xml2dict) + + +class convertsizeTests(TestCaseBase): + def testBytesToMB(self): + retVal = utils.convertSize(778999, 'B', 'MB') + assert (round(retVal, 2) == 0.74) + + def testPBtoMB(self): + retVal = utils.convertSize(8, 'PB', 'MB') + assert (round(retVal, 0) == 8589934592) + + def testInvalidInput(self): + with self.assertRaises(ValueError): + utils.convertSize(89, 'P', 'M') + + def testMBToMB(self): + retVal = utils.convertSize(900, 'MB', 'MB') + assert (retVal == 900) -- cgit