message: Switch VariantTest to use TF combinations
diff: @@ -17,16 +17,18 @@ from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  +from absl.testing import parameterized
  +
  from tensorflow.python.data.experimental.ops import cardinality
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.data.ops import dataset_ops
  -from tensorflow.python.framework import test_util
  +from tensorflow.python.framework import combinations
  from tensorflow.python.platform import test
  -@test_util.run_all_in_graph_and_eager_modes
  -class VariantTest(test_base.DatasetTestBase):
  +class VariantTest(test_base.DatasetTestBase, parameterized.TestCase):
  + @combinations.generate(test_base.default_test_combinations())
  def testRoundtripRange(self):
  dataset = dataset_ops.Dataset.range(10)
  variant = dataset_ops.to_variant(dataset)
  @@ -35,6 +37,7 @@ class VariantTest(test_base.DatasetTestBase):
  self.assertDatasetProduces(dataset, range(10))
  self.assertEqual(self.evaluate(cardinality.cardinality(dataset)), 10)
  + @combinations.generate(test_base.default_test_combinations())
  def testRoundtripMap(self):
  dataset = dataset_ops.Dataset.range(10).map(lambda x: x*x)
  variant = dataset_ops.to_variant(dataset)
  