ENH: arange default dtype

Author: ev-br

array_api_strict/_creation_functions.py

@@ -127,6 +127,11 @@ def arange(
 
     _check_device(device)
     _check_valid_dtype(dtype, device)
+    if dtype is None:
+        if any(isinstance(x, float) for x in (start, stop, step)):
+            dtype = get_default_dtypes(device)["real floating"]
+        else:
+            dtype = get_default_dtypes(device)["integral"]
 
     return Array._new(
         np.arange(start, stop, step, dtype=_np_dtype(dtype)),

array_api_strict/tests/test_creation_functions.py

@@ -281,7 +281,6 @@ def test_eye(self):
         with pytest.raises((TypeError, ValueError)):
             eye(3, device=device, dtype=float64)
 
-
     def test_linspace(self):
         device = Device('F32_device')
 
@@ -294,11 +293,23 @@ def test_linspace(self):
         with pytest.raises((TypeError, ValueError)):
             linspace(1, 10, 11, device=device, dtype=float64)
 
+    def test_arange(self):
+        device = Device('F32_device')
+
+        a = arange(0, 10, 1, device=device)
+        assert a.dtype == self.info.default_dtypes(device=device)["integral"]
+
+        a = arange(0.0, 10, 1, device=device)
+        assert a.dtype == self.info.default_dtypes(device=device)["real floating"]
+
+        with pytest.raises((TypeError, ValueError)):
+            arange(0, 10, 1, device=device, dtype=float64)
 
+        with pytest.raises((TypeError, ValueError)):
+            arange(0.0, 10, 1, device=device, dtype=float64)
 
 # TODO:
 # def asarray(
-# def arange(
 
 
 @pytest.mark.parametrize("api_version", ['2021.12', '2022.12', '2023.12'])