aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/data/experimental/kernel_tests/shuffle_and_repeat_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/data/experimental/kernel_tests/shuffle_and_repeat_test.py')
-rw-r--r--tensorflow/python/data/experimental/kernel_tests/shuffle_and_repeat_test.py21
1 files changed, 1 insertions, 20 deletions
diff --git a/tensorflow/python/data/experimental/kernel_tests/shuffle_and_repeat_test.py b/tensorflow/python/data/experimental/kernel_tests/shuffle_and_repeat_test.py
index 883169495f..c208963a86 100644
--- a/tensorflow/python/data/experimental/kernel_tests/shuffle_and_repeat_test.py
+++ b/tensorflow/python/data/experimental/kernel_tests/shuffle_and_repeat_test.py
@@ -17,7 +17,6 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
-from absl.testing import parameterized
import numpy as np
from tensorflow.python.data.experimental.ops import shuffle_ops
@@ -28,7 +27,7 @@ from tensorflow.python.framework import ops
from tensorflow.python.platform import test
-class ShuffleAndRepeatTest(test_base.DatasetTestBase, parameterized.TestCase):
+class ShuffleAndRepeatTest(test_base.DatasetTestBase):
def _build_ds(self, seed, count=5, num_elements=20):
return dataset_ops.Dataset.range(num_elements).apply(
@@ -111,24 +110,6 @@ class ShuffleAndRepeatTest(test_base.DatasetTestBase, parameterized.TestCase):
with self.session(graph=g) as sess:
sess.run(get_next_op)
- @parameterized.named_parameters(
- ("NoSeed", None),
- ("WithSeed", 42),
- )
- def testShuffleAndRepeatAndZipDataset(self, seed):
- dataset = dataset_ops.Dataset.range(10).apply(
- shuffle_ops.shuffle_and_repeat(10, count=3, seed=seed))
- dataset = dataset_ops.Dataset.zip((dataset, dataset))
- iterator = dataset.make_one_shot_iterator()
- next_element = iterator.get_next()
-
- with self.cached_session() as sess:
- for _ in range(30):
- x, y = sess.run(next_element)
- self.assertEqual(x, y)
- with self.assertRaises(errors.OutOfRangeError):
- sess.run(next_element)
-
if __name__ == "__main__":
test.main()