MAINT: dedupe real/complex helpers

Author: ev-br

array_api_tests/dtype_helpers.py

@@ -231,26 +231,6 @@ def real_dtype_for(dtyp):
     return real_dtype
 
 
-def complex_for_float(dtyp):
-    """For a real or complex dtype, return a matching complex dtype."""
-    if api_version <= '2021.12':
-        raise TypeError("complex dtypes require api_version >= 2022.12.")
-
-    if dtyp not in all_float_dtypes:
-        raise ValueError(f"expected a real dtype, got {dtyp}.")
-
-    if dtyp == xp.float32:
-        return xp.complex64
-    elif dtyp == xp.float64:
-        return xp.complex128
-    elif dtyp == xp.complex64:
-        return xp.complex64
-    elif dtype == xp.complex128:
-        return xp.complex128
-    else:
-        raise ValueError(f"Unknown dtype {dtyp}.")
-
-
 def _make_dtype_mapping_from_names(mapping: Dict[str, Any]) -> EqualityMapping:
     dtype_value_pairs = []
     for name, value in mapping.items():

array_api_tests/test_linalg.py

@@ -343,7 +343,7 @@ def test_eig(x):
 
     eigenvalues = res.eigenvalues
     eigenvectors = res.eigenvectors
-    expected_dtype = dh.complex_for_float(x.dtype)
+    expected_dtype = dh.complex_dtype_for(x.dtype)
 
     ph.assert_dtype("eig", in_dtype=x.dtype, out_dtype=eigenvalues.dtype,
                     expected=expected_dtype, repr_name="eigenvalues.dtype")
@@ -373,7 +373,7 @@ def test_eig(x):
 @given(x=arrays(dtype=all_floating_dtypes(), shape=square_matrix_shapes))
 def test_eigvals(x):
     res = linalg.eigvals(x)
-    expected_dtype = dh.complex_for_float(x.dtype)
+    expected_dtype = dh.complex_dtype_for(x.dtype)
 
     ph.assert_dtype("eigvals", in_dtype=x.dtype, out_dtype=res.dtype,
                     expected=expected_dtype, repr_name="eigvals")