array-api-strict 2.0__tar.gz → 2.0.1__tar.gz

{array_api_strict-2.0 → array_api_strict-2.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: array_api_strict
-Version: 2.0
+Version: 2.0.1
 Summary: A strict, minimal implementation of the Python array API standard.
 Home-page: https://data-apis.org/array-api-strict/
 Author: Consortium for Python Data API Standards

{array_api_strict-2.0 → array_api_strict-2.0.1}/array_api_strict/_array_object.py

@@ -677,10 +677,16 @@ class Array:
         """
         Performs the operation __iter__.
         """
-        # Manually disable iteration, since __getitem__ raises IndexError on
-        # things like ones((3, 3))[0], which causes list(ones((3, 3))) to give
-        # [].
-        raise TypeError("array iteration is not allowed in array-api-strict")
+        # 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) for i in self._array)
 
     def __le__(self: Array, other: Union[int, float, Array], /) -> Array:
         """

{array_api_strict-2.0 → array_api_strict-2.0.1}/array_api_strict/_flags.py

@@ -81,8 +81,8 @@ def set_array_api_strict_flags(
         array-api-strict will change the default version to 2023.12.
 
     boolean_indexing : bool, optional
-        Whether indexing by a boolean array is supported.
-        Note that although boolean array indexing does result in
+        Whether indexing by a boolean array is supported. This flag is enabled
+        by default. Note that although boolean array indexing does result in
         data-dependent shapes, this flag is independent of the
         `data_dependent_shapes` flag (see below).
 
@@ -276,6 +276,19 @@ class ArrayAPIStrictFlags:
     See :func:`set_array_api_strict_flags` for a
     description of the available flags.
 
+    Examples
+    --------
+
+    >>> from array_api_strict import ArrayAPIStrictFlags, get_array_api_strict_flags
+    >>> with ArrayAPIStrictFlags(api_version="2022.12", boolean_indexing=False):
+    ...     flags = get_array_api_strict_flags()
+    >>> flags
+    {'api_version': '2022.12',
+     'boolean_indexing': False,
+     'data_dependent_shapes': True,
+     'enabled_extensions': ('linalg', 'fft')
+    }
+
     See Also
     --------
 

{array_api_strict-2.0 → array_api_strict-2.0.1}/array_api_strict/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2024-06-27T16:40:56-0600",
+ "date": "2024-07-01T13:26:20-0600",
  "dirty": false,
  "error": null,
- "full-revisionid": "eb8dd464eea5873a59f73c109d8e3347918319d7",
- "version": "2.0"
+ "full-revisionid": "7c55e4712b706128366a12fc3bea82d182b9f1a6",
+ "version": "2.0.1"
 }
 '''  # END VERSION_JSON
 

{array_api_strict-2.0 → array_api_strict-2.0.1}/array_api_strict/tests/test_array_object.py

@@ -1,4 +1,5 @@
 import operator
+from builtins import all as all_
 
 from numpy.testing import assert_raises, suppress_warnings
 import numpy as np
@@ -21,6 +22,7 @@ from .._dtypes import (
     int32,
     int64,
     uint64,
+    float64,
     bool as bool_,
 )
 from .._flags import set_array_api_strict_flags
@@ -423,8 +425,12 @@ def test_array_namespace():
     pytest.raises(ValueError, lambda: a.__array_namespace__(api_version="2021.11"))
     pytest.raises(ValueError, lambda: a.__array_namespace__(api_version="2024.12"))
 
-def test_no_iter():
-    pytest.raises(TypeError, lambda: iter(ones(3)))
+def test_iter():
+    pytest.raises(TypeError, lambda: iter(asarray(3)))
+    assert list(ones(3)) == [asarray(1.), asarray(1.), asarray(1.)]
+    assert all_(isinstance(a, Array) for a in iter(ones(3)))
+    assert all_(a.shape == () for a in iter(ones(3)))
+    assert all_(a.dtype == float64 for a in iter(ones(3)))
     pytest.raises(TypeError, lambda: iter(ones((3, 3))))
 
 @pytest.mark.parametrize("api_version", ['2021.12', '2022.12', '2023.12'])

{array_api_strict-2.0 → array_api_strict-2.0.1}/array_api_strict.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: array_api_strict
-Version: 2.0
+Version: 2.0.1
 Summary: A strict, minimal implementation of the Python array API standard.
 Home-page: https://data-apis.org/array-api-strict/
 Author: Consortium for Python Data API Standards