Test failure with Python 3.14 beta 1

[frenzymadness]

According to this record in the changelog of Python 3.14 beta 1:

gh-127682: No longer call iter twice when creating and executing a generator expression. Creating a generator expression from a non-interable will raise only when the generator expression is executed. This brings the behavior of generator expressions in line with other generators.

the expected exception in:

array-api-strict/array_api_strict/tests/test_array_object.py

Line 640 in e25b928

pytest.raises(TypeError, lambda: iter(asarray(3)))

is not gonna be raised until the generator returned from the __iter__ method of Array object is accessed. Adding next(…) around the iter(…) call makes the test compatible with both 3.14b1 and older Pythons.

I'm testing this with older version of the package so I'm sorry if it's no longer relevant. I see the __iter__ code is slightly different in the latest version.

[ev-br]

Interesting. Would you be interested in checking if the problem also exists on main, and sending a PR with a fix if it does?

[frenzymadness]

It seems the problem is in the main branch as well:

<mock-chroot> sh-5.2# grep -A 15 "def __iter__" array_api_strict/_array_object.py 
    def __iter__(self) -> Iterator[Array]:
        """
        Performs the operation __iter__.
        """
        # Manually disable iteration on higher dimensional arrays, since
        # __getitem__ raises IndexError on things like ones((3, 3))[0], which
        # causes list(ones((3, 3))) to give [].
        if self.ndim > 1:
            raise TypeError("array iteration is not allowed in array-api-strict")
        # Allow iteration for 1-D arrays. The array API doesn't strictly
        # define __iter__, but it doesn't disallow it. The default Python
        # behavior is to implement iter as a[0], a[1], ... when __getitem__ is
        # implemented, which implies iteration on 1-D arrays.
        return (Array._new(i, device=self.device) for i in self._array)

    def __le__(self, other: Array | float, /) -> Array:


<mock-chroot> sh-5.2# python3
Python 3.14.0b1 (main, May  7 2025, 00:00:00) [GCC 15.1.1 20250425 (Red Hat 15.1.1-1)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from array_api_strict import asarray
>>> iter(asarray(3))
<generator object Array.__iter__.<locals>.<genexpr> at 0x7ff2165464d0>
>>> next(iter(asarray(3)))
Traceback (most recent call last):
  File "<python-input-2>", line 1, in <module>
    next(iter(asarray(3)))
    ~~~~^^^^^^^^^^^^^^^^^^
  File "/builddir/array-api-strict/array_api_strict/_array_object.py", line 794, in <genexpr>
    return (Array._new(i, device=self.device) for i in self._array)
                                                       ^^^^^^^^^^^
TypeError: iteration over a 0-d array
>>>

I can send a PR fixing the test by adding the call to next if tha's the fix you've meant.

[ev-br]

Yes please, it'd be great!