Add support for setting the api version to 2023.12

Author: asmeurer

array_api_strict/_flags.py

@@ -21,6 +21,7 @@
 supported_versions = (
     "2021.12",
     "2022.12",
+    "2023.12",
 )
 
 API_VERSION = default_version = "2022.12"
@@ -67,6 +68,9 @@ def set_array_api_strict_flags(
       Note that 2021.12 is supported, but currently gives the same thing as
       2022.12 (except that the fft extension will be disabled).
 
+      2023.12 support is preliminary. Some features in 2023.12 may still be
+      missing, and it hasn't been fully tested.
+
     - `data_dependent_shapes`: Whether data-dependent shapes are enabled in
       array-api-strict.
 
@@ -123,6 +127,8 @@ def set_array_api_strict_flags(
             raise ValueError(f"Unsupported standard version {api_version!r}")
         if api_version == "2021.12":
             warnings.warn("The 2021.12 version of the array API specification was requested but the returned namespace is actually version 2022.12")
+        if api_version == "2023.12":
+            warnings.warn("The 2023.12 version of the array API specification is still preliminary. Some functions are not yet implemented, and it has not been fully tested.")
         API_VERSION = api_version
         array_api_strict.__array_api_version__ = API_VERSION
 

array_api_strict/tests/test_array_object.py

@@ -410,9 +410,12 @@ def test_array_namespace():
     assert a.__array_namespace__(api_version="2022.12") is array_api_strict
     assert array_api_strict.__array_api_version__ == "2022.12"
 
+    assert a.__array_namespace__(api_version="2023.12") is array_api_strict
+    assert array_api_strict.__array_api_version__ == "2023.12"
+
     with pytest.warns(UserWarning):
         assert a.__array_namespace__(api_version="2021.12") is array_api_strict
     assert array_api_strict.__array_api_version__ == "2021.12"
 
     pytest.raises(ValueError, lambda: a.__array_namespace__(api_version="2021.11"))
-    pytest.raises(ValueError, lambda: a.__array_namespace__(api_version="2023.12"))
+    pytest.raises(ValueError, lambda: a.__array_namespace__(api_version="2024.12"))

array_api_strict/tests/test_flags.py

@@ -54,6 +54,19 @@ def test_flags():
         'data_dependent_shapes': True,
         'enabled_extensions': ('linalg',),
     }
+    reset_array_api_strict_flags()
+
+    # 2023.12 should issue a warning
+    with pytest.warns(UserWarning) as record:
+        set_array_api_strict_flags(api_version='2023.12')
+    assert len(record) == 1
+    assert '2023.12' in str(record[0].message)
+    flags = get_array_api_strict_flags()
+    assert flags == {
+        'api_version': '2023.12',
+        'data_dependent_shapes': True,
+        'enabled_extensions': ('linalg', 'fft'),
+    }
 
     # Test setting flags with invalid values
     pytest.raises(ValueError, lambda: