TST: adapt tests for the lack of __array__

Author: ev-br

array_api_strict/_creation_functions.py

@@ -3,7 +3,7 @@
 from collections.abc import Generator
 from contextlib import contextmanager
 from enum import Enum
-from typing import TYPE_CHECKING, Literal
+from typing import TYPE_CHECKING, Literal, List, Optional, Tuple, Union
 
 import numpy as np
 
@@ -26,20 +26,6 @@ class Undef(Enum):
 _undef = Undef.UNDEF
 
 
-@contextmanager
-def allow_array() -> Generator[None]:
-    """
-    Temporarily enable Array.__array__. This is needed for np.array to parse
-    list of lists of Array objects.
-    """
-    from . import _array_object
-    original_value = _array_object._allow_array
-    try:
-        _array_object._allow_array = True
-        yield
-    finally:
-        _array_object._allow_array = original_value
-
 
 def _check_valid_dtype(dtype: DType | None) -> None:
     # Note: Only spelling dtypes as the dtype objects is supported.
@@ -123,8 +109,8 @@ def asarray(
         # Give a better error message in this case. NumPy would convert this
         # to an object array. TODO: This won't handle large integers in lists.
         raise OverflowError("Integer out of bounds for array dtypes")
-    with allow_array():
-        res = np.array(obj, dtype=_np_dtype, copy=copy)
+
+    res = np.array(obj, dtype=_np_dtype, copy=copy)
     return Array._new(res, device=device)
 
 

array_api_strict/tests/test_array_object.py

@@ -1,3 +1,4 @@
+import sys
 import operator
 from builtins import all as all_
 
@@ -526,6 +527,10 @@ def test_array_properties():
     assert b.mT.shape == (3, 2)
 
 
+@pytest.mark.xfail(sys.version_info.major*100 + sys.version_info.minor < 312,
+                   reason="array conversion relies on buffer protocol, and "
+                          "requires python >= 3.12"
+)
 def test_array_conversion():
     # Check that arrays on the CPU device can be converted to NumPy
     # but arrays on other devices can't. Note this is testing the logic in
@@ -539,22 +544,6 @@ def test_array_conversion():
         with pytest.raises(RuntimeError, match="Can not convert array"):
             np.asarray(a)
 
-def test__array__():
-    # __array__ should work for now
-    a = ones((2, 3))
-    np.array(a)
-
-    # Test the _allow_array private global flag for disabling it in the
-    # future.
-    from .. import _array_object
-    original_value = _array_object._allow_array
-    try:
-        _array_object._allow_array = False
-        a = ones((2, 3))
-        with pytest.raises(ValueError, match="Conversion from an array_api_strict array to a NumPy ndarray is not supported"):
-            np.array(a)
-    finally:
-        _array_object._allow_array = original_value
 
 def test_allow_newaxis():
     a = ones(5)

array_api_strict/tests/test_creation_functions.py

@@ -1,3 +1,4 @@
+import sys
 import warnings
 
 from numpy.testing import assert_raises
@@ -97,7 +98,6 @@ def test_asarray_copy():
     a[0] = 0
     assert all(b[0] == 0)
 
-
 def test_asarray_list_of_lists():
     lst = [[1, 2, 3], [4, 5, 6]]
     res = asarray(lst)