message: use combinations in traverse_test
diff: @@ -20,7 +20,8 @@ from __future__ import print_function
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.util import traverse
  -from tensorflow.python.framework import test_util
  +from tensorflow.python.data.kernel_tests import test_base
  +from tensorflow.python.framework import combinations
  from tensorflow.python.ops import gen_dataset_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
  @@ -41,13 +42,13 @@ class _TestDataset(dataset_ops.UnaryUnchangedStructureDataset):
  class TraverseTest(test.TestCase):
  - @test_util.run_deprecated_v1
  + @combinations.generate(test_base.graph_only_combinations())
  def testOnlySource(self):
  ds = dataset_ops.Dataset.range(10)
  variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)
  self.assertAllEqual(["RangeDataset"], [x.name for x in variant_tensor_ops])
  - @test_util.run_deprecated_v1
  + @combinations.generate(test_base.graph_only_combinations())
  def testSimplePipeline(self):
  ds = dataset_ops.Dataset.range(10).map(math_ops.square)
  variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)
  @@ -55,7 +56,7 @@ class TraverseTest(test.TestCase):
  set(["MapDataset", "RangeDataset"]),
  set(x.name for x in variant_tensor_ops))
  - @test_util.run_deprecated_v1
  + @combinations.generate(test_base.graph_only_combinations())
  def testConcat(self):
  ds1 = dataset_ops.Dataset.range(10)
  ds2 = dataset_ops.Dataset.range(10)
  @@ -65,7 +66,7 @@ class TraverseTest(test.TestCase):
  set(["ConcatenateDataset", "RangeDataset", "RangeDataset_1"]),
  set(x.name for x in variant_tensor_ops))
  - @test_util.run_deprecated_v1
  + @combinations.generate(test_base.graph_only_combinations())
  def testZip(self):
  ds1 = dataset_ops.Dataset.range(10)
  ds2 = dataset_ops.Dataset.range(10)
  @@ -75,7 +76,7 @@ class TraverseTest(test.TestCase):
  set(["ZipDataset", "RangeDataset", "RangeDataset_1"]),
  set(x.name for x in variant_tensor_ops))
  - @test_util.run_deprecated_v1
  + @combinations.generate(test_base.graph_only_combinations())
  def testMultipleVariantTensors(self):
  ds = dataset_ops.Dataset.range(10)
  ds = _TestDataset(ds)
  @@ -84,7 +85,7 @@ class TraverseTest(test.TestCase):
  set(["RangeDataset", "ModelDataset", "PrefetchDataset"]),
  set(x.name for x in variant_tensor_ops))
  - @test_util.run_deprecated_v1
  + @combinations.generate(test_base.graph_only_combinations())
  def testFlatMap(self):
  ds1 = dataset_ops.Dataset.range(10).repeat(10)
  