diff options
| author | Raghuveer Devulapalli <raghuveer.devulapalli@intel.com> | 2021-10-19 12:40:08 -0700 |
|---|---|---|
| committer | Raghuveer Devulapalli <raghuveer.devulapalli@intel.com> | 2022-01-14 10:28:39 -0800 |
| commit | b96d2e430a42ef48447f29bdfad1c3ba3cca980f (patch) | |
| tree | ffc0e947623336ae9e33da590569ff7b1c119088 /numpy | |
| parent | 8c5e7e64f08025120dc73c95e43af3b95a4c49a6 (diff) | |
| download | numpy-b96d2e430a42ef48447f29bdfad1c3ba3cca980f.tar.gz | |
TST: Use default_rng to generate random numbers for the test
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/core/tests/test_multiarray.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py index f0db82e1b..73bb5e2d8 100644 --- a/numpy/core/tests/test_multiarray.py +++ b/numpy/core/tests/test_multiarray.py @@ -9309,11 +9309,11 @@ def test_sort_float(N): def test_sort_int(): # Random data with NPY_MAX_INT32 and NPY_MIN_INT32 sprinkled - np.random.seed(42) + rng = np.random.default_rng(42) N = 2047 minv = np.iinfo(np.int32).min maxv = np.iinfo(np.int32).max - arr = np.random.randint(low=minv, high=maxv, size=10).astype('int32') + arr = rng.integers(low=minv, high=maxv, size=N).astype('int32') arr[np.random.choice(arr.shape[0], 10)] = minv arr[np.random.choice(arr.shape[0], 10)] = maxv assert_equal(np.sort(arr, kind='quick'), np.sort(arr, kind='heap')) @@ -9321,9 +9321,9 @@ def test_sort_int(): def test_sort_uint(): # Random data with NPY_MAX_UINT32 sprinkled - np.random.seed(42) + rng = np.random.default_rng(42) N = 2047 maxv = np.iinfo(np.uint32).max - arr = np.random.randint(low=0, high=maxv, size=10).astype('uint32') + arr = rng.integers(low=0, high=maxv, size=N).astype('uint32') arr[np.random.choice(arr.shape[0], 10)] = maxv assert_equal(np.sort(arr, kind='quick'), np.sort(arr, kind='heap')) |