import unittest

from hysteria_panel import utils


class TestUtils(unittest.TestCase):
    def test_format_bytes(self):
        self.assertEqual(utils.format_bytes(0), "0 B")
        self.assertEqual(utils.format_bytes(1), "1.00 B")
        self.assertEqual(utils.format_bytes(1024), "1.00 KB")
        self.assertEqual(utils.format_bytes(1024**2), "1.00 MB")
        self.assertEqual(utils.format_bytes(1536), "1.50 KB")

    def test_format_speed(self):
        self.assertEqual(utils.format_speed(0), "0 B/s")
        self.assertEqual(utils.format_speed(1), "1.00 B/s")
        self.assertEqual(utils.format_speed(1024), "1.00 KB/s")
        self.assertEqual(utils.format_speed(1024**2), "1.00 MB/s")
        self.assertEqual(utils.format_speed(1536), "1.50 KB/s")


if __name__ == "__main__":
    unittest.main()