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)