Fix test condition for choice method

cupy · Jun 5, 2017 · 96b6fd7 · 96b6fd7
1 parent fdda76e
commit 96b6fd7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/cupy_tests/random_tests/test_generator.py b/tests/cupy_tests/random_tests/test_generator.py
@@ -332,9 +332,9 @@ def test_goodness_of_fit_2(self):
 @testing.gpu
 class TestChoiceMultinomial(unittest.TestCase):
 
-    @condition.retry(5)
+    @condition.retry(10)
     @testing.for_float_dtypes()
-    @testing.numpy_cupy_allclose(atol=0.01)
+    @testing.numpy_cupy_allclose(atol=0.02)
     def test_choice_multinomial(self, xp, dtype):
         p = xp.array([0.2, 0.3, 0.5], dtype)
         trial = 10000