Use fixed number to test

2020-12-08 20:53:39 -08:00 · 2020-12-08 20:53:39 -08:00 · 620fa97685
commit 620fa97685
parent a49b21b06e
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/python/kernel_tests/unique_op_test.py
+++ b/tensorflow/python/kernel_tests/unique_op_test.py
@ -110,7 +110,7 @@ class UniqueTest(test.TestCase):
  @test_util.run_deprecated_v1
  def testShapeInferenceV2(self):
    """Test shape inference."""
-    x = np.random.randint(2, high=10, size=(3, 2, 1))
+    x = np.arange(6).reshape(3, 2, 1)
    _, idx = gen_array_ops.unique_v2(x, axis=[0])
    self.assertEqual(idx.shape.as_list(), [3])
    _, idx = gen_array_ops.unique_v2(x, axis=[1])