Temporary workaround for problem with 'multivariate' when specifying zero

probabilities.

1 files changed, 13 insertions, 0 deletions

diff --git a/numpy/random/tests/test_random.py b/numpy/random/tests/test_random.py
new file mode 100644
index 000000000..8f7842085
--- /dev/null
+++ b/numpy/random/tests/test_random.py
@@ -0,0 +1,13 @@
+from numpy.testing import *
+from numpy import random
+import numpy as N
+
+class test_multinomial(NumpyTestCase):
+    def test_basic(self):
+        random.multinomial(100, [0.2, 0.8])
+
+    def test_zero_probability(self):
+        random.multinomial(100, [0.2, 0.8, 0.0, 0.0, 0.0])
+
+if __name__ == "__main__":
+    NumpyTest().run()