Skip to content

Commit eb6330f

Browse files
Changed test special cases for log1p to work with NumPy 2.0
1 parent dc26e2e commit eb6330f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dpctl/tests/elementwise/test_log1p.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,4 +164,6 @@ def test_log1p_special_cases():
164164

165165
tol = dpt.finfo(X.dtype).resolution
166166
with np.errstate(invalid="ignore"):
167-
assert_allclose(dpt.asnumpy(dpt.log1p(X)), res, atol=tol, rtol=tol)
167+
dpt_res = dpt.asnumpy(dpt.log1p(X))
168+
assert_allclose(np.real(dpt_res), np.real(res), atol=tol, rtol=tol)
169+
assert_allclose(np.imag(dpt_res), np.imag(res), atol=tol, rtol=tol)

0 commit comments

Comments
 (0)