def add_size_to_holders_tree(tree):
"""add size information to generated holders trees"""
size_file = os.path.join(tree['device'], 'size')
# size file is always represented in 512 byte sectors even if
# underlying disk uses a larger logical_block_size
size = ((512 * int(util.load_file(size_file)))
if os.path.exists(size_file) else None)
tree['size'] = util.bytes2human(size) if args.human else str(size)
for holder in tree['holders']:
add_size_to_holders_tree(holder)
return tree
def test_b2h_values(self):
self.assertEqual('10G', util.bytes2human(10 * self.GB))
self.assertEqual('10M', util.bytes2human(10 * self.MB))
self.assertEqual('1000B', util.bytes2human(1000))
self.assertEqual('1K', util.bytes2human(1024))
self.assertEqual('1K', util.bytes2human(1024.0))
self.assertEqual('1T', util.bytes2human(float(1024 * self.GB)))
def test_h2b_b2b(self):
for size_str in ['10G', '20G', '2T', '12K', '1M', '1023K']:
self.assertEqual(util.bytes2human(util.human2bytes(size_str)),
size_str)