Skip to content

Commit c8c9498

Browse files
committed
Enable dtype checks in sum and prod for 2023.12
1 parent 5e62058 commit c8c9498

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

array_api_tests/test_statistical_functions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def test_prod(x, data):
148148
# See https://github.com/data-apis/array-api-tests/issues/106
149149
if x.dtype in dh.uint_dtypes:
150150
assert dh.is_int_dtype(out.dtype) # sanity check
151-
elif api_version < "2023.12": # TODO: update dtype assertion for >2023.12 - see #234
151+
else:
152152
ph.assert_dtype("prod", in_dtype=x.dtype, out_dtype=out.dtype, expected=expected_dtype)
153153
_axes = sh.normalise_axis(kw.get("axis", None), x.ndim)
154154
ph.assert_keepdimable_shape(
@@ -237,7 +237,7 @@ def test_sum(x, data):
237237
# See https://github.com/data-apis/array-api-tests/issues/160
238238
if x.dtype in dh.uint_dtypes:
239239
assert dh.is_int_dtype(out.dtype) # sanity check
240-
elif api_version < "2023.12": # TODO: update dtype assertion for >2023.12 - see #234
240+
else:
241241
ph.assert_dtype("sum", in_dtype=x.dtype, out_dtype=out.dtype, expected=expected_dtype)
242242
_axes = sh.normalise_axis(kw.get("axis", None), x.ndim)
243243
ph.assert_keepdimable_shape(

0 commit comments

Comments
 (0)