message: Add enumerate test on Dataset
diff: @@ -24,9 +24,11 @@ import six
  from tensorflow.python.autograph.operators import data_structures
  from tensorflow.python.autograph.operators import py_builtins
  +from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  +from tensorflow.python.framework import tensor_shape
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import tensor_array_ops
  @@ -142,6 +144,25 @@ class PyBuiltinsTest(test.TestCase):
  self.assertListEqual(list(py_builtins.enumerate_([3,2,1], 5)), [(5, 3), (6, 2), (7, 1)])
  self.assertListEqual(list(py_builtins.enumerate_([-8], -3)), [(-3, -8)])
  + @test_util.run_all_in_graph_and_eager_modes
  + def test_enumerate_dataset(self):
  + components = (["a", "b"], [1, 2], [37.0, 38])
  + start = constant_op.constant(20, dtype=dtypes.int64)
  +
  + dataset = py_builtins.enumerate_(dataset_ops.Dataset.from_tensor_slices(
  + components), start)
  +
  + self.assertEqual(dtypes.int64,
  + dataset_ops.get_legacy_output_types(dataset)[0])
  + dataset_output_shapes = dataset_ops.get_legacy_output_shapes(dataset)
  + self.assertEqual((), dataset_output_shapes[0])
  + self.assertEqual([tensor_shape.TensorShape([])] * 3,
  + [shape for shape in dataset_output_shapes[1]])
  +
  + self.assertDatasetProduces(dataset, [(20, (b"a", 1, 37.0)),
  + (21, (b"b", 2, 38.0))])
  +
  +
  def test_eval_in_original_context(self):
  def caller_1(lvl_delta):
  