diff options
Diffstat (limited to 'tensorflow/python/training/input_test.py')
-rw-r--r-- | tensorflow/python/training/input_test.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/python/training/input_test.py b/tensorflow/python/training/input_test.py index 4f705a0a85..d32768f5da 100644 --- a/tensorflow/python/training/input_test.py +++ b/tensorflow/python/training/input_test.py @@ -429,6 +429,13 @@ class DictHelperTest(test_lib.TestCase): d2 = inp._as_original_type(d, l) self.assertEquals(d, d2) + def testHeterogeneousKeysDictInputs(self): + d = {"z": 1, 1: 42, ("a", "b"): 100} + l = inp._as_tensor_list(d) + self.assertEquals([100, 42, 1], l) + d2 = inp._as_original_type(d, l) + self.assertEquals(d, d2) + class BatchTest(test_lib.TestCase): |