未验证 提交 2d275451 编写于 作者: A Amador Pahim

Merge branch 'clebergnu-data_size_int'

Signed-off-by: NAmador Pahim <apahim@redhat.com>
......@@ -297,8 +297,8 @@ class DataSize(object):
"""
if attr not in self.MULTIPLIERS:
raise AttributeError('Attribute %s does not exist.' % attr)
return (self.value * self.MULTIPLIERS[self.unit] /
self.MULTIPLIERS[attr])
return int(self.value * self.MULTIPLIERS[self.unit] /
self.MULTIPLIERS[attr])
def __dir__(self):
"""
......
......@@ -105,9 +105,11 @@ class TestDataSize(unittest.TestCase):
self.assertRaises(data_structures.InvalidDataSize,
data_structures.DataSize, '10Mb')
def test_value_and_type(self):
self.assertIs(data_structures.DataSize('0b').b, 0)
self.assertIs(data_structures.DataSize('0t').b, 0)
def test_values(self):
self.assertEqual(data_structures.DataSize('0b').b, 0)
self.assertEqual(data_structures.DataSize('0t').b, 0)
self.assertEqual(data_structures.DataSize('10m').b, 10485760)
self.assertEqual(data_structures.DataSize('10M').b, 10485760)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册