diff options
Diffstat (limited to 'numpy/core')
| -rw-r--r-- | numpy/core/src/private/templ_common.h.src | 9 | ||||
| -rw-r--r-- | numpy/core/tests/test_multiarray.py | 21 |
2 files changed, 25 insertions, 5 deletions
diff --git a/numpy/core/src/private/templ_common.h.src b/numpy/core/src/private/templ_common.h.src index dd6c7bf23..a65a00758 100644 --- a/numpy/core/src/private/templ_common.h.src +++ b/numpy/core/src/private/templ_common.h.src @@ -17,6 +17,10 @@ /* * writes result of a * b into r * returns 1 if a * b overflowed else returns 0 + * + * These functions are not designed to work if either a or b is negative, but + * that is not checked. Could use absolute values and adjust the sign if that + * functionality was desired. */ static NPY_INLINE int npy_mul_with_overflow_@name@(@type@ * r, @type@ a, @type@ b) @@ -24,17 +28,16 @@ npy_mul_with_overflow_@name@(@type@ * r, @type@ a, @type@ b) #ifdef HAVE___BUILTIN_MUL_OVERFLOW return __builtin_mul_overflow(a, b, r); #else - const @type@ half_sz = (((@type@)1 << (sizeof(a) * 8 / 2)) - 1); + const @type@ half_sz = ((@type@)1 << ((sizeof(a) * 8 - 1 ) / 2)); *r = a * b; /* * avoid expensive division on common no overflow case */ if (NPY_UNLIKELY((a | b) >= half_sz) && - a != 0 && b > @MAX@ / a) { + a != 0 && b > @MAX@ / a) { return 1; } - return 0; #endif } diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py index f5991d052..dd51077b9 100644 --- a/numpy/core/tests/test_multiarray.py +++ b/numpy/core/tests/test_multiarray.py @@ -564,6 +564,21 @@ class TestCreation(TestCase): arr = np.array([], dtype='V') assert_equal(arr.dtype.kind, 'V') + def test_too_big_error(self): + # 45341 is the smallest integer greater than sqrt(2**31 - 1). + # 3037000500 is the smallest integer greater than sqrt(2**63 - 1). + # We want to make sure that the square byte array with those dimensions + # is too big on 32 or 64 bit systems respectively. + if np.iinfo('intp').max == 2**31 - 1: + shape = (46341, 46341) + elif np.iinfo('intp').max == 2**63 - 1: + shape = (3037000500, 3037000500) + else: + return + assert_raises(ValueError, np.empty, shape, dtype=np.int8) + assert_raises(ValueError, np.zeros, shape, dtype=np.int8) + assert_raises(ValueError, np.ones, shape, dtype=np.int8) + def test_zeros(self): types = np.typecodes['AllInteger'] + np.typecodes['AllFloat'] for dt in types: @@ -5996,8 +6011,10 @@ class TestArrayAttributeDeletion(object): """ticket #2046, should not seqfault, raise AttributeError""" a = np.ones(2) attr = ['shape', 'strides', 'data', 'dtype', 'real', 'imag', 'flat'] - for s in attr: - assert_raises(AttributeError, delattr, a, s) + with warnings.catch_warnings(): + warnings.simplefilter('ignore') + for s in attr: + assert_raises(AttributeError, delattr, a, s) def test_multiarray_not_writable_attributes_deletion(self): a = np.ones(2) |