Skip to content

Commit d952af1

Browse files
use deterministic seeds in tests (uxlfoundation#2230)
1 parent d9cd092 commit d952af1

File tree

2 files changed

+6
-10
lines changed

2 files changed

+6
-10
lines changed

onedal/utils/tests/test_validation.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,6 @@
1414
# limitations under the License.
1515
# ==============================================================================
1616

17-
import time
18-
1917
import numpy as np
2018
import numpy.random as rand
2119
import pytest
@@ -65,7 +63,7 @@ def test_sum_infinite_actually_finite(dtype, shape, allow_nan, dataframe, queue)
6563
)
6664
@pytest.mark.parametrize("allow_nan", [False, True])
6765
@pytest.mark.parametrize("check", ["inf", "NaN", None])
68-
@pytest.mark.parametrize("seed", [0, int(time.time())])
66+
@pytest.mark.parametrize("seed", [0, 123456])
6967
@pytest.mark.parametrize(
7068
"dataframe, queue", get_dataframes_and_queues("numpy,dpnp,dpctl")
7169
)
@@ -92,7 +90,7 @@ def test_assert_finite_random_location(
9290
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
9391
@pytest.mark.parametrize("allow_nan", [False, True])
9492
@pytest.mark.parametrize("check", ["inf", "NaN", None])
95-
@pytest.mark.parametrize("seed", [0, int(time.time())])
93+
@pytest.mark.parametrize("seed", [0, 123456])
9694
@pytest.mark.parametrize(
9795
"dataframe, queue", get_dataframes_and_queues("numpy,dpnp,dpctl")
9896
)
@@ -120,7 +118,7 @@ def test_assert_finite_random_shape_and_location(
120118
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
121119
@pytest.mark.parametrize("allow_nan", [False, True])
122120
@pytest.mark.parametrize("check", ["inf", "NaN", None])
123-
@pytest.mark.parametrize("seed", [0, int(time.time())])
121+
@pytest.mark.parametrize("seed", [0, 123456])
124122
def test_assert_finite_sparse(dtype, allow_nan, check, seed):
125123
lb, ub = 2, 2056
126124
rand.seed(seed)

sklearnex/utils/tests/test_validation.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,6 @@
1414
# limitations under the License.
1515
# ==============================================================================
1616

17-
import time
18-
1917
import numpy as np
2018
import numpy.random as rand
2119
import pytest
@@ -67,7 +65,7 @@ def test_sum_infinite_actually_finite(dtype, shape, ensure_all_finite):
6765
)
6866
@pytest.mark.parametrize("ensure_all_finite", ["allow-nan", True])
6967
@pytest.mark.parametrize("check", ["inf", "NaN", None])
70-
@pytest.mark.parametrize("seed", [0, int(time.time())])
68+
@pytest.mark.parametrize("seed", [0, 123456])
7169
@pytest.mark.parametrize(
7270
"dataframe, queue",
7371
get_dataframes_and_queues(_dataframes_supported),
@@ -110,7 +108,7 @@ def test_validate_data_random_location(
110108
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
111109
@pytest.mark.parametrize("ensure_all_finite", ["allow-nan", True])
112110
@pytest.mark.parametrize("check", ["inf", "NaN", None])
113-
@pytest.mark.parametrize("seed", [0, int(time.time())])
111+
@pytest.mark.parametrize("seed", [0, 123456])
114112
@pytest.mark.parametrize(
115113
"dataframe, queue",
116114
get_dataframes_and_queues(_dataframes_supported),
@@ -151,7 +149,7 @@ def test_validate_data_random_shape_and_location(
151149

152150
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
153151
@pytest.mark.parametrize("check", ["inf", "NaN", None])
154-
@pytest.mark.parametrize("seed", [0, int(time.time())])
152+
@pytest.mark.parametrize("seed", [0, 123456])
155153
@pytest.mark.parametrize(
156154
"dataframe, queue",
157155
get_dataframes_and_queues(_dataframes_supported),

0 commit comments

Comments
 (0)