From 0256451181b8c415bd18a8196fc87d5957d4db05 Mon Sep 17 00:00:00 2001 From: Oleksandr Pavlyk Date: Mon, 16 Sep 2024 07:31:00 -0500 Subject: [PATCH] Undo workaround to verify the fix --- mkl_random/tests/test_random.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/mkl_random/tests/test_random.py b/mkl_random/tests/test_random.py index abb311a..557474a 100644 --- a/mkl_random/tests/test_random.py +++ b/mkl_random/tests/test_random.py @@ -804,8 +804,7 @@ def test_randomdist_normal(randomdist): np.testing.assert_allclose(actual, desired, atol=1e-8, rtol=1e-8) rnd.seed(randomdist.seed, brng=randomdist.brng) - workaround = rnd.normal(loc=.123456789, scale=2.0, size=(4, 2), method="BoxMuller2") - actual = workaround[:3,:] + actual = rnd.normal(loc=.123456789, scale=2.0, size=(6, 2), method="BoxMuller2") desired = np.array([[0.16673479781277187, 0.48153966449249175], [-3.4809986872165952, -0.8101190082826486], [-0.051937610825354905, 2.4088402362484342]]) @@ -903,8 +902,7 @@ def test_randomdist_standard_normal(randomdist): np.testing.assert_allclose(actual, desired, atol=1e-7, rtol=1e-10) rnd.seed(randomdist.seed, brng=randomdist.brng) - workaround = rnd.standard_normal(size=(4, 2), method='BoxMuller2') - actual = workaround[:3, :] + actual = rnd.standard_normal(size=(3, 2), method='BoxMuller2') desired = np.array([[0.021639004406385935, 0.17904143774624587], [-1.8022277381082976, -0.4667878986413243], [-0.08769719991267745, 1.1426917236242171]])