import unittest
from unit import to_byte


class TestUnit(unittest.TestCase):

    def test_to_byte(self):
        self.assertTrue(to_byte("1b"), 1)
        self.assertTrue(to_byte("1K"), 1024)
        self.assertNotEqual(to_byte("1g"), 1024*1024*102)
if __name__ == "__main__":
    unittest.main()