with seed for 2 random large array tests

apache · Feb 24, 2019 · 482fdef · 482fdef
1 parent 8818dea
commit 482fdef
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
@@ -18,6 +18,7 @@
 import mxnet as mx
 import numpy as np
 from mxnet import gluon, nd
+from tests.python.unittest.common import with_seed
 
 # dimension constants
 MEDIUM_X = 10000
@@ -45,10 +46,12 @@ def test_ndarray_ones():
     assert a[-1][0] == 1
     assert nd.sum(a).asnumpy() == LARGE_SIZE
 
+@with_seed()
 def test_ndarray_random_uniform():
     a = nd.random.uniform(shape=(LARGE_X, SMALL_Y))
     assert a[-1][0] != 0
 
+@with_seed()
 def test_ndarray_random_randint():
     a = nd.random.randint(100, 10000, shape=(LARGE_X, SMALL_Y))
     assert a.shape == (LARGE_X, SMALL_Y)