Jan 20, 2024 · Nov 10, 2023 · Dec 17, 2023 · Dec 17, 2023 · Dec 17, 2023 · Dec 17, 2023
diff --git a/numpy/_core/tests/test_umath.py b/numpy/_core/tests/test_umath.py
        class StoreArrayPrepareWrap(np.ndarray):
            _wrap_args = None
            _prepare_args = None

            def __new__(cls):
                return np.zeros(()).view(cls)

            def __array_wrap__(self, obj, context, return_scalar):
                self._wrap_args = context[1]
                return obj

            @property
            def args(self):
                # We need to ensure these are fetched at the same time, before
                # any other ufuncs are called by the assertions
                return self._wrap_args

            def __repr__(self):
                return "a"  # for short test output
Original file line number	Diff line number	Diff line change
Expand Up		@@ -3006,16 +3006,20 @@ def test_wrap_out(self):
		class StoreArrayPrepareWrap(np.ndarray):
		_wrap_args = None
		_prepare_args = None

		def __new__(cls):
		return np.zeros(()).view(cls)

		def __array_wrap__(self, obj, context, return_scalar):
		self._wrap_args = context[1]
		return obj

		@property
		def args(self):
		# We need to ensure these are fetched at the same time, before
		# any other ufuncs are called by the assertions
		return self._wrap_args

		def __repr__(self):
		return "a" # for short test output

Expand Down