Don't re-enable disabled extensions when setting the api version

Author: asmeurer

array_api_strict/_flags.py

@@ -140,7 +140,7 @@ def set_array_api_strict_flags(
                 )
         ENABLED_EXTENSIONS = tuple(enabled_extensions)
     else:
-        ENABLED_EXTENSIONS = tuple([ext for ext in all_extensions if extension_versions[ext] <= API_VERSION])
+        ENABLED_EXTENSIONS = tuple([ext for ext in ENABLED_EXTENSIONS if extension_versions[ext] <= API_VERSION])
 
 # We have to do this separately or it won't get added as the docstring
 set_array_api_strict_flags.__doc__ = set_array_api_strict_flags.__doc__.format(

array_api_strict/tests/test_flags.py

@@ -42,6 +42,16 @@ def test_flags():
     assert flags == {
         'api_version': '2021.12',
         'data_dependent_shapes': False,
+        'enabled_extensions': (),
+    }
+    reset_array_api_strict_flags()
+
+    with pytest.warns(UserWarning):
+        set_array_api_strict_flags(api_version='2021.12')
+    flags = get_array_api_strict_flags()
+    assert flags == {
+        'api_version': '2021.12',
+        'data_dependent_shapes': True,
         'enabled_extensions': ('linalg',),
     }