Skip to content

Commit 0154f9c

Browse files
committed
Avoid overflow in windows in test
np.arange creates a numpy array with long int elements which is 32 bit on windows.
1 parent e465ad2 commit 0154f9c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

symengine/tests/test_lambdify.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -493,9 +493,9 @@ def test_more_than_255_args():
493493
input_arr = np.arange(q, q + n*n).reshape((n, n))
494494
out = callback(input_arr)
495495
ref = np.empty((n, 3))
496-
coeffs = np.arange(p, p + n)
496+
coeffs = np.arange(p, p + n, dtype=np.int64)
497497
for i in range(n):
498-
ref[i, 0] = coeffs.dot(np.arange(q + n*i, q + n*(i+1)))
498+
ref[i, 0] = coeffs.dot(np.arange(q + n*i, q + n*(i+1), dtype=np.int64))
499499
ref[i, 1] = q + n*i + r
500500
ref[:, 2] = -99
501501
assert np.allclose(out, ref)

0 commit comments

Comments
 (0)