Undo workaround to verify the fix

IntelPython · Sep 16, 2024 · 0256451 · 0256451
1 parent e1ed130
commit 0256451
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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]])