def test_get_name_reduction():
# Existing reduction
reduction = UnitByte.get_name_reduction(UnitByte.megabyte)
assert reduction is 'MB'
# Non-existing reduction
reduction = UnitByte.get_name_reduction('Non-existing reduction')
assert reduction is None
def byte_value(value):
formatted_value = ''
if isinstance(value, (int, float)):
value, unit = UnitByte.auto_convert(value)
value = '{:.2f}'.format(value)
unit = UnitByte.get_name_reduction(unit)
formatted_value = value + unit
return formatted_value
def _format_bytes(self, byte_value):
try:
if (byte_value is None) or (byte_value == 0):
byte_value = '0'
elif isinstance(byte_value, (int, float)):
val, unit = UnitByte.auto_convert(byte_value)
byte_value = '{:.2f}'.format(val) + \
UnitByte.get_name_reduction(unit)
finally:
return byte_value
def _format_bytes(self, byte_value):
try:
if (byte_value is None) or (byte_value == 0):
byte_value = '0'
elif isinstance(byte_value, (int, float)):
val, unit = UnitByte.auto_convert(byte_value)
byte_value = '{:.2f}'.format(val) + \
UnitByte.get_name_reduction(unit)
finally:
return byte_value
