diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py index 69fe3bc2bae4..a627467cb959 100644 --- 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)