def test_func_kwargs_params (self): @baker.command(shortopts={"verbose": "v"}, params={"verbose": "Be verbose"}) def func(*args, **kwargs): return kwargs['verbose'] res = baker.run(argv=[__file__, 'func', '--verbose', 'arg1']) self.assertEqual(res, True) res = baker.run(argv=[__file__, 'func', 'arg1', '--verbose']) self.assertEqual(res, True) res = baker.run(argv=[__file__, 'func', '-v']) self.assertEqual(res, True) res = baker.run(argv=[__file__, 'func', '-v', 'arg1']) self.assertEqual(res, True)
def test_func_noargs (self): @baker.command def func(): return 42 argv = [__file__, 'func'] self.assertEqual(baker.test(argv=argv), "func()") self.assertEqual(baker.run(argv=argv), 42)
def test_func_kwargs_revorder (self): @baker.command def func(arg1, arg2, *args, **kwargs): return arg1, arg2, kwargs['verbose'] argv = [__file__, 'func', 'argvalue1', '--verbose', 'argvalue2'] self.assertEqual(baker.test(argv=argv), "func('argvalue1','argvalue2',verbose=True)") self.assertEqual(baker.run(argv=argv), ('argvalue1', 'argvalue2', True))
def test_func_args (self): @baker.command def func(a, b, c): return (a, b, c) argv = [__file__, 'func', '1', '2', '3'] self.assertEqual(baker.test(argv=argv), "func('1','2','3')") self.assertEqual(baker.run(argv=argv), ('1','2','3'))