from unittest import TestCase

from bauh.commons import system


class GetHumanSizeStr(TestCase):

    def test__must_properly_convert_to_Kb(self):
        self.assertEqual('1.00 Kb', system.get_human_size_str(1024))
        self.assertEqual('1.50 Kb', system.get_human_size_str(1536))
        self.assertEqual('1.75 Kb', system.get_human_size_str(1792))
        self.assertEqual('57.30 Kb', system.get_human_size_str(58675.2))

    def test__must_properly_convert_to_Mb(self):
        self.assertEqual('1.00 Mb', system.get_human_size_str(1048576))
        self.assertEqual('1.20 Mb', system.get_human_size_str(1258291))
        self.assertEqual('1.50 Mb', system.get_human_size_str(1572864))
        self.assertEqual('57.30 Mb', system.get_human_size_str(60083404.8))

    def test__must_properly_convert_to_Gb(self):
        self.assertEqual('1.00 Gb', system.get_human_size_str(1073741824))
        self.assertEqual('57.30 Gb', system.get_human_size_str(61525406515.2))

    def test__must_properly_convert_to_Tb(self):
        self.assertEqual('1.00 Tb', system.get_human_size_str(1099511627776))
        self.assertEqual('57.30 Tb', system.get_human_size_str(63002016271564.8))

    def test__must_properly_convert_to_Pb(self):
        self.assertEqual('1.00 Pb', system.get_human_size_str(1125899906842624))
        self.assertEqual('57.30 Pb', system.get_human_size_str(64514064662082350))