ENH: eye default dtype

Author: ev-br

array_api_strict/_creation_functions.py

@@ -193,6 +193,8 @@ def eye(
 
     _check_device(device)
     _check_valid_dtype(dtype, device)
+    if dtype is None:
+        dtype = get_default_dtypes(device)["real floating"]
 
     return Array._new(
         np.eye(n_rows, M=n_cols, k=k, dtype=_np_dtype(dtype)), device=device

array_api_strict/tests/test_creation_functions.py

@@ -273,15 +273,20 @@ def test_ones_like_etc_incorrect(self, func):
         with pytest.raises((TypeError, ValueError)):
             func(a, device=Device('F32_device'), dtype=float64)
 
+    def test_eye(self):
+        device = Device('F32_device')
+        a = eye(3, device=device)
+        assert a.dtype == self.info.default_dtypes(device=device)["real floating"]
+
+        with pytest.raises((TypeError, ValueError)):
+            eye(3, device=device, dtype=float64)
+
 
 # TODO:
 # def asarray(
 # def arange(
-# def eye(
 # def linspace(
-# def meshgrid(*arrays: Array, indexing: Literal["xy", "ij"] = "xy") -> tuple[Array, ...]:
-# def tril(x: Array, /, *, k: int = 0) -> Array:
-# def triu(x: Array, /, *, k: int = 0) -> Array:
+
 
 
 @pytest.mark.parametrize("api_version", ['2021.12', '2022.12', '2023.12'])