From 121fcf2cb0c77f7ca257b4dac488a6814fd89e74 Mon Sep 17 00:00:00 2001 From: Pradyot Ranjan <99216956+pradyotRanjan@users.noreply.github.com> Date: Sat, 6 Jun 2026 03:26:02 +0530 Subject: [PATCH] Better error message Signed-off-by: Pradyot Ranjan <99216956+pradyotRanjan@users.noreply.github.com> --- src/array_api_extra/testing.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/array_api_extra/testing.py b/src/array_api_extra/testing.py index 9166dfd1..24b2a0f5 100644 --- a/src/array_api_extra/testing.py +++ b/src/array_api_extra/testing.py @@ -770,6 +770,7 @@ def assert_close( Array arguments to `atol` and `rtol` must be valid input to :class:`float`. """ + __tracebackhide__ = True actual, desired, xp, np = _check_ns_shape_dtype( actual, desired, check_dtype, check_shape, check_scalar, xp ) @@ -850,6 +851,7 @@ def assert_equal( assert_close : Similar function for inexact equality checks. numpy.testing.assert_array_equal : Similar function for NumPy arrays. """ + __tracebackhide__ = True actual, desired, xp, np = _check_ns_shape_dtype( actual, desired, check_dtype, check_shape, check_scalar, xp ) @@ -908,6 +910,7 @@ def assert_less( assert_close : Similar function for inexact equality checks. numpy.testing.assert_array_less : Similar function for NumPy arrays. """ + __tracebackhide__ = True x, y, xp, np = _check_ns_shape_dtype( x, y, check_dtype, check_shape, check_scalar, xp )