def test_use_option_parser_the_right_way(self): sys.argv = ['wrapping.py', '-s', '1', '-t', 'DBNet', '-o', 'SMAC'] args, unknown = wrapping.use_arg_parser() self.assertEqual(args.optimizer, 'SMAC') self.assertEqual(args.seed, 1) self.assertEqual(args.title, 'DBNet') self.assertEqual(len(unknown), 0)
def test_parse_config_values_from_unknown_arguments2(self): """Test if we can convert a config with Sections and variables into an argparser. Test for arguments with whitespaces""" sys.argv = ['wrapping.py', '-s', '1', '-t', 'DBNet', '-o', 'SMAC', '--HPOLIB:function', 'python', '../branin.py'] args, unknown = wrapping.use_arg_parser() self.assertEqual(len(unknown), 3) config = ConfigParser.SafeConfigParser(allow_no_value=True) config.read("dummy_config.cfg") config_args = wrapping_util.parse_config_values_from_unknown_arguments( unknown, config) self.assertListEqual(vars(config_args)['HPOLIB:function'], ['python', '../branin.py']) self.assertIs(vars(config_args)['GRIDSEARCH:params'], None) self.assertIs(vars(config_args)['HPOLIB:result_on_terminate'], None)