diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-27 08:02:42 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 08:06:22 -0700 |
| commit | 40a36ea2bd415f498baf3ec065a274ba8e712e84 (patch) | |
| tree | a81a9398303ca2e71569114406a476c5ede0c3ac /tensorflow/python/util | |
| parent | 3d30dd424c0404ea5349c0d2acdde2acd4e0aa97 (diff) | |
Update HasKwargsTest ensuring that internal checks for tests involving functools.partial are triggered.
PiperOrigin-RevId: 214775194
Diffstat (limited to 'tensorflow/python/util')
| -rw-r--r-- | tensorflow/python/util/function_utils_test.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/python/util/function_utils_test.py b/tensorflow/python/util/function_utils_test.py index ce768637f5..e5b0843e4b 100644 --- a/tensorflow/python/util/function_utils_test.py +++ b/tensorflow/python/util/function_utils_test.py @@ -183,6 +183,8 @@ class HasKwargsTest(test.TestCase): wrapped_fn = functools.partial(fn_has_kwargs, test_arg=123) self.assertTrue(function_utils.has_kwargs(wrapped_fn)) + some_kwargs = dict(x=1, y=2, z=3) + self.assertEqual(wrapped_fn(**some_kwargs), some_kwargs) def fn_has_no_kwargs(x, test_arg): if test_arg != expected_test_arg: @@ -191,6 +193,8 @@ class HasKwargsTest(test.TestCase): wrapped_fn = functools.partial(fn_has_no_kwargs, test_arg=123) self.assertFalse(function_utils.has_kwargs(wrapped_fn)) + some_arg = 1 + self.assertEqual(wrapped_fn(some_arg), some_arg) def test_double_partial(self): expected_test_arg1 = 123 @@ -205,6 +209,8 @@ class HasKwargsTest(test.TestCase): double_wrapped_fn = functools.partial(wrapped_fn, test_arg1=123) self.assertTrue(function_utils.has_kwargs(double_wrapped_fn)) + some_kwargs = dict(x=1, y=2, z=3) + self.assertEqual(double_wrapped_fn(**some_kwargs), some_kwargs) def fn_has_no_kwargs(x, test_arg1, test_arg2): if test_arg1 != expected_test_arg1 or test_arg2 != expected_test_arg2: @@ -215,6 +221,8 @@ class HasKwargsTest(test.TestCase): double_wrapped_fn = functools.partial(wrapped_fn, test_arg1=123) self.assertFalse(function_utils.has_kwargs(double_wrapped_fn)) + some_arg = 1 + self.assertEqual(double_wrapped_fn(some_arg), some_arg) def test_raises_type_error(self): with self.assertRaisesRegexp( |