def test_cmd_args_with_vars(self): auto_var = AutoVar(logging_level=logging.INFO) auto_var.add_variable_class(DatasetVarClass()) args = ['--dataset', 'halfmoon_300'] auto_var.parse_argparse(args=args) with self.assertRaises(argparse.ArgumentTypeError): args = ['--dataset', 'halfmaan_300'] auto_var.parse_argparse(args=args) with self.assertRaises(SystemExit) as cm: args = [] auto_var.parse_argparse(args=args) self.assertEqual(cm.exception.code, 2)
def test_cmd_args(self): auto_var = AutoVar(logging_level=logging.INFO) self.assertFalse(auto_var._no_hooks) args = ['--no-hooks'] auto_var.parse_argparse(args=args) self.assertTrue(auto_var._no_hooks)
def test_cmd_args_with_vars_2(self): auto_var = AutoVar(logging_level=logging.INFO) auto_var.add_variable_class(OrdVarClass()) auto_var.parse_argparse(args=[]) self.assertEqual(auto_var.get_var("ord"), 2)