MAINT: drop ceil/trunc/floor from common/_aliases.py

Author: ev-br

array_api_compat/common/_aliases.py

@@ -524,33 +524,6 @@ def nonzero(x: Array, /, xp: Namespace, **kwargs: object) -> tuple[Array, ...]:
     return xp.nonzero(x, **kwargs)
 
 
-# ceil, floor, and trunc return integers for integer inputs
-
-
-def ceil(x: Array, /, xp: Namespace, **kwargs: object) -> Array:
-    result = xp.ceil(x, **kwargs)
-    if result.dtype != x.dtype:
-        # numpy < 2: ceil(int array) is float
-        result = xp.asarray(result, dtype=x.dtype)
-    return result
-
-
-def floor(x: Array, /, xp: Namespace, **kwargs: object) -> Array:
-    result = xp.floor(x, **kwargs)
-    if result.dtype != x.dtype:
-        # numpy < 2: floor(int array) is float
-        result = xp.asarray(result, dtype=x.dtype)
-    return result
-
-
-def trunc(x: Array, /, xp: Namespace, **kwargs: object) -> Array:
-    result = xp.trunc(x, **kwargs)
-    if result.dtype != x.dtype:
-        # numpy < 2: trunc(int array) is float
-        result = xp.asarray(result, dtype=x.dtype)
-    return result
-
-
 # linear algebra functions
 
 
@@ -713,9 +686,6 @@ def iinfo(type_: DType | Array, /, xp: Namespace) -> Any:
     "argsort",
     "sort",
     "nonzero",
-    "ceil",
-    "floor",
-    "trunc",
     "matmul",
     "matrix_transpose",
     "tensordot",

array_api_compat/cupy/_aliases.py

@@ -54,9 +54,6 @@
 argsort = get_xp(cp)(_aliases.argsort)
 sort = get_xp(cp)(_aliases.sort)
 nonzero = get_xp(cp)(_aliases.nonzero)
-ceil = get_xp(cp)(_aliases.ceil)
-floor = get_xp(cp)(_aliases.floor)
-trunc = get_xp(cp)(_aliases.trunc)
 matmul = get_xp(cp)(_aliases.matmul)
 matrix_transpose = get_xp(cp)(_aliases.matrix_transpose)
 tensordot = get_xp(cp)(_aliases.tensordot)

array_api_compat/dask/array/_aliases.py

@@ -134,9 +134,6 @@ def arange(
 matrix_transpose = get_xp(da)(_aliases.matrix_transpose)
 vecdot = get_xp(da)(_aliases.vecdot)
 nonzero = get_xp(da)(_aliases.nonzero)
-ceil = get_xp(np)(_aliases.ceil)
-floor = get_xp(np)(_aliases.floor)
-trunc = get_xp(np)(_aliases.trunc)
 matmul = get_xp(np)(_aliases.matmul)
 tensordot = get_xp(np)(_aliases.tensordot)
 sign = get_xp(np)(_aliases.sign)

array_api_compat/numpy/_aliases.py

@@ -63,9 +63,6 @@
 argsort = get_xp(np)(_aliases.argsort)
 sort = get_xp(np)(_aliases.sort)
 nonzero = get_xp(np)(_aliases.nonzero)
-ceil = get_xp(np)(_aliases.ceil)
-floor = get_xp(np)(_aliases.floor)
-trunc = get_xp(np)(_aliases.trunc)
 matmul = get_xp(np)(_aliases.matmul)
 matrix_transpose = get_xp(np)(_aliases.matrix_transpose)
 tensordot = get_xp(np)(_aliases.tensordot)

tests/test_copies_or_views.py

@@ -37,7 +37,7 @@ def is_view(func, a, value):
     """Apply `func`, mutate the output; does the input change?"""
     b = func(a)
     b[0] = value
-    return a[0] == value
+    return a[0] == value, b.dtype == a.dtype
 
 
 @pytest.mark.parametrize('xp_name', LIB_NAMES) 
@@ -57,10 +57,11 @@ def test_view_or_copy(inputs, xp_name):
 
     # bare namespace: mutate the output, does the input change?
     a = bare_xp.asarray(arr_input, dtype=dtype)
-    is_view_bare = is_view(bare_func, a, value)
+    is_view_bare, dtype_same_bare = is_view(bare_func, a, value)
 
     # wrapped namespace: mutate the output, does the input change?
     a1 = wrapped_xp.asarray(arr_input, dtype=dtype)
-    is_view_wrapped = is_view(wrapped_func, a1, value)
+    is_view_wrapped, dtype_same_wrapped = is_view(wrapped_func, a1, value)
 
     assert is_view_bare == is_view_wrapped
+    assert dtype_same_bare == dtype_same_wrapped