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)