message: Switch UniqueTest to use TF combinations
diff: @@ -17,17 +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 unique
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.data.ops import dataset_ops
  +from tensorflow.python.framework import combinations
  from tensorflow.python.framework import dtypes
  -from tensorflow.python.framework import test_util
  from tensorflow.python.platform import test
  from tensorflow.python.util import compat
  -@test_util.run_all_in_graph_and_eager_modes
  -class UniqueTest(test_base.DatasetTestBase):
  +class UniqueTest(test_base.DatasetTestBase, parameterized.TestCase):
  def _testSimpleHelper(self, dtype, test_cases):
  """Test the `unique()` transformation on a list of test cases.
  @@ -52,7 +53,8 @@ class UniqueTest(test_base.DatasetTestBase):
  for element in expected
  ])
  - @test_util.run_deprecated_v1
  + @combinations.generate(
  + combinations.combine(tf_api_version=[1], mode=["graph", "eager"]))
  def testSimpleInt(self):
  for dtype in [dtypes.int32, dtypes.int64]:
  self._testSimpleHelper(dtype, [
  @@ -65,7 +67,8 @@ class UniqueTest(test_base.DatasetTestBase):
  ([[1, 1], [1, 1], [2, 2], [3, 3], [1, 1]], [[1, 1], [2, 2], [3, 3]]),
  ])
  - @test_util.run_deprecated_v1
  + @combinations.generate(
  + combinations.combine(tf_api_version=[1], mode=["graph", "eager"]))
  def testSimpleString(self):
  self._testSimpleHelper(dtypes.string, [
  ([], []),
  