From 6edbd6b7d14ae81e359b328a47d6c6187d0d5844 Mon Sep 17 00:00:00 2001 From: Ryan Sepassi Date: Fri, 3 Aug 2018 11:29:38 -0700 Subject: [PATCH] Remove flaky testNoShuffleFail test that was testing non-guaranteed behavior PiperOrigin-RevId: 207297174 --- tensor2tensor/data_generators/problem_test.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/tensor2tensor/data_generators/problem_test.py b/tensor2tensor/data_generators/problem_test.py index 5e76db9e8..26e370bd8 100644 --- a/tensor2tensor/data_generators/problem_test.py +++ b/tensor2tensor/data_generators/problem_test.py @@ -73,18 +73,6 @@ def testNoShufflePreprocess(self): with tf.Session() as sess: self.assertTrue(assert_tensors_equal(sess, tensor1, tensor2, 20)) - def testNoShuffleFail(self): - problem = algorithmic.TinyAlgo() - dataset = problem.dataset(mode=tf.estimator.ModeKeys.TRAIN, - data_dir=algorithmic.TinyAlgo.data_dir, - shuffle_files=True) - - tensor1 = dataset.make_one_shot_iterator().get_next()["targets"] - tensor2 = dataset.make_one_shot_iterator().get_next()["targets"] - - with tf.Session() as sess: - self.assertFalse(assert_tensors_equal(sess, tensor1, tensor2, 20)) - if __name__ == "__main__": tf.test.main()