def test_cli_script_main_from_and_to_unit(self): """CLI script returns correct if given FROM and TO units""" args = ['-f', 'MiB', '-t', 'Byte', '1'] # Testing FROM 1 MiB TO equivalent Bytes results = bitmath.cli_script_main(args) self.assertEqual(results[0], bitmath.Byte(1048576)) self.assertIs(type(results[0]), bitmath.Byte)
def test_cli_script_main_from_unit(self): """CLI script returns correct if given FROM units""" args = ['-f', 'MiB', '0.5'] # Testing FROM 0.5 MiB TO best human readable unit (512 KiB) results = bitmath.cli_script_main(args) self.assertEqual(results[0], bitmath.KiB(512)) self.assertIs(type(results[0]), bitmath.KiB)
def test_cli_script_main_to_unit(self): """CLI script returns correct TO units""" args = ['-t', 'MiB', '1048576'] results = bitmath.cli_script_main(args) self.assertEqual(results[0], bitmath.MiB(1)) self.assertIs(type(results[0]), bitmath.MiB)
def test_cli_script_main_no_units(self): """CLI script works if no to/from units are provided""" args = ['100', '1024'] results = bitmath.cli_script_main(args) self.assertEqual(results[0], bitmath.Byte(100)) self.assertEqual(results[1], bitmath.KiB(1))
def test_cli_script_main_no_options(self): """CLI script returns nothing if no input is given""" results = bitmath.cli_script_main([]) self.assertEqual(results, [])