Skip to content

Commit 47697e3

Browse files
Implement work-around for the test failure due to a bug in MKL
1 parent 9640607 commit 47697e3

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

mkl_random/tests/test_random.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -804,7 +804,8 @@ def test_randomdist_normal(randomdist):
804804
np.testing.assert_allclose(actual, desired, atol=1e-8, rtol=1e-8)
805805

806806
rnd.seed(randomdist.seed, brng=randomdist.brng)
807-
actual = rnd.normal(loc=.123456789, scale=2.0, size=(3, 2), method="BoxMuller2")
807+
workaround = rnd.normal(loc=.123456789, scale=2.0, size=(4, 2), method="BoxMuller2")
808+
actual = workaround[:3,:]
808809
desired = np.array([[0.16673479781277187, 0.48153966449249175],
809810
[-3.4809986872165952, -0.8101190082826486],
810811
[-0.051937610825354905, 2.4088402362484342]])
@@ -902,7 +903,8 @@ def test_randomdist_standard_normal(randomdist):
902903
np.testing.assert_allclose(actual, desired, atol=1e-7, rtol=1e-10)
903904

904905
rnd.seed(randomdist.seed, brng=randomdist.brng)
905-
actual = rnd.standard_normal(size=(3, 2), method='BoxMuller2')
906+
workaround = rnd.standard_normal(size=(4, 2), method='BoxMuller2')
907+
actual = workaround[:3, :]
906908
desired = np.array([[0.021639004406385935, 0.17904143774624587],
907909
[-1.8022277381082976, -0.4667878986413243],
908910
[-0.08769719991267745, 1.1426917236242171]])

0 commit comments

Comments
 (0)