From 89ecc557fc9205e06c9816ce2122f42e3adced02 Mon Sep 17 00:00:00 2001 From: Kevin Sheppard Date: Tue, 14 Jul 2020 17:55:54 +0100 Subject: BUG: Validate output size in bin- and multinomial Check that size is not being broadcast closes #16833 --- numpy/random/_common.pxd | 2 ++ 1 file changed, 2 insertions(+) (limited to 'numpy/random/_common.pxd') diff --git a/numpy/random/_common.pxd b/numpy/random/_common.pxd index 588f613ae..4f404b7a1 100644 --- a/numpy/random/_common.pxd +++ b/numpy/random/_common.pxd @@ -77,6 +77,8 @@ cdef object wrap_int(object val, object bits) cdef np.ndarray int_to_array(object value, object name, object bits, object uint_size) +cdef validate_output_shape(iter_shape, np.ndarray output) + cdef object cont(void *func, void *state, object size, object lock, int narg, object a, object a_name, constraint_type a_constraint, object b, object b_name, constraint_type b_constraint, -- cgit v1.2.1