self.assertIn('<code>test_function', docs)
def test_argspec_for_functools_partial(self):
-
# pylint: disable=unused-argument
def test_function_for_partial1(arg1, arg2, kwarg1=1, kwarg2=2):
pass
# pylint: disable=protected-access
# Make sure everything works for regular functions.
- expected = tf_inspect.ArgSpec(['arg1', 'arg2', 'kwarg1', 'kwarg2'], None,
- None, (1, 2))
+ expected = tf_inspect.FullArgSpec(args=['arg1', 'arg2', 'kwarg1', 'kwarg2'],
+ varargs=None, varkw=None, defaults=(1, 2),
+ kwonlyargs=[], kwonlydefaults={},
+ annotations={})
self.assertEqual(expected, parser._get_arg_spec(test_function_for_partial1))
# Make sure doing nothing works.
- expected = tf_inspect.ArgSpec(['arg1', 'arg2', 'kwarg1', 'kwarg2'], None,
- None, (1, 2))
+ expected = tf_inspect.FullArgSpec(args=['arg1', 'arg2', 'kwarg1', 'kwarg2'],
+ varargs=None, varkw=None, defaults=(1, 2),
+ kwonlyargs=[], kwonlydefaults={},
+ annotations={})
partial = functools.partial(test_function_for_partial1)
self.assertEqual(expected, parser._get_arg_spec(partial))
# Make sure setting args from the front works.
- expected = tf_inspect.ArgSpec(['arg2', 'kwarg1', 'kwarg2'], None, None,
- (1, 2))
+ expected = tf_inspect.FullArgSpec(args=['arg2', 'kwarg1', 'kwarg2'],
+ varargs=None, varkw=None, defaults=(1, 2),
+ kwonlyargs=[], kwonlydefaults={},
+ annotations={})
partial = functools.partial(test_function_for_partial1, 1)
self.assertEqual(expected, parser._get_arg_spec(partial))
- expected = tf_inspect.ArgSpec(['kwarg2',], None, None, (2,))
+ expected = tf_inspect.FullArgSpec(args=['kwarg2'],
+ varargs=None, varkw=None, defaults=(2,),
+ kwonlyargs=[], kwonlydefaults={},
+ annotations={})
partial = functools.partial(test_function_for_partial1, 1, 2, 3)
self.assertEqual(expected, parser._get_arg_spec(partial))
# Make sure setting kwargs works.
- expected = tf_inspect.ArgSpec(['arg1', 'arg2', 'kwarg2'], None, None, (2,))
+ expected = tf_inspect.FullArgSpec(args=['arg1', 'arg2', 'kwarg2'],
+ varargs=None, varkw=None, defaults=(2,),
+ kwonlyargs=[], kwonlydefaults={},
+ annotations={})
partial = functools.partial(test_function_for_partial1, kwarg1=0)
self.assertEqual(expected, parser._get_arg_spec(partial))
- expected = tf_inspect.ArgSpec(['arg1', 'arg2', 'kwarg1'], None, None, (1,))
+ expected = tf_inspect.FullArgSpec(args=['arg1', 'arg2', 'kwarg1'],
+ varargs=None, varkw=None, defaults=(1,),
+ kwonlyargs=[], kwonlydefaults={},
+ annotations={})
partial = functools.partial(test_function_for_partial1, kwarg2=0)
self.assertEqual(expected, parser._get_arg_spec(partial))
- expected = tf_inspect.ArgSpec(['arg1'], None, None, ())
+ expected = tf_inspect.FullArgSpec(args=['arg1'],
+ varargs=None, varkw=None, defaults=(),
+ kwonlyargs=[], kwonlydefaults={},
+ annotations={})
partial = functools.partial(test_function_for_partial1,
arg2=0, kwarg1=0, kwarg2=0)
self.assertEqual(expected, parser._get_arg_spec(partial))
# Make sure *args, *kwargs is accounted for.
- expected = tf_inspect.ArgSpec([], 'my_args', 'my_kwargs', ())
+ expected = tf_inspect.FullArgSpec(args=[],
+ varargs='my_args', varkw='my_kwargs',
+ defaults=(),
+ kwonlyargs=[], kwonlydefaults={},
+ annotations={})
partial = functools.partial(test_function_for_partial2, 0, 1)
self.assertEqual(expected, parser._get_arg_spec(partial))
-
+
# pylint: enable=protected-access
def testSaveReferenceResolver(self):