fix more tests

Author: lithomas1

array_api_compat/dask/array/_aliases.py

@@ -103,6 +103,15 @@ def dask_arange(
 matrix_transpose = get_xp(da)(_aliases.matrix_transpose)
 vecdot = get_xp(da)(_aliases.vecdot)
 
+nonzero = get_xp(da)(_aliases.nonzero)
+sum = get_xp(np)(_aliases.sum)
+prod = get_xp(np)(_aliases.prod)
+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)
+
 from dask.array import (
     # Element wise aliases
     arccos as acos,
@@ -120,9 +129,17 @@ def dask_arange(
     concatenate as concat,
 )
 
-del da, partial
+# exclude these from all since
+_da_unsupported = ['sort', 'argsort']
+
+common_aliases = [alias for alias in _aliases.__all__ if alias not in _da_unsupported]
+
+__all__ = common_aliases + ['asarray', 'bool', 'acos',
+                            'acosh', 'asin', 'asinh', 'atan', 'atan2',
+                            'atanh', 'bitwise_left_shift', 'bitwise_invert',
+                            'bitwise_right_shift', 'concat', 'pow',
+                            'e', 'inf', 'nan', 'pi', 'newaxis', 'float32', 'float64', 'int8',
+                            'int16', 'int32', 'int64', 'uint8', 'uint16', 'uint32', 'uint64',
+                            'complex64', 'complex128', 'iinfo', 'finfo', 'can_cast', 'result_type']
 
-__all__ = _aliases.__all__ + ['asarray', 'bool', 'acos',
-                              'acosh', 'asin', 'asinh', 'atan', 'atan2',
-                              'atanh', 'bitwise_left_shift', 'bitwise_invert',
-                              'bitwise_right_shift', 'concat', 'pow']
+del da, partial, common_aliases, _da_unsupported,

dask-xfails.txt

@@ -59,16 +59,6 @@ array_api_tests/test_has_names.py::test_has_names[array_method-to_device]
 array_api_tests/test_has_names.py::test_has_names[array_attribute-device]
 array_api_tests/test_has_names.py::test_has_names[array_attribute-mT]
 
-# dask doesn't return int when input is already int for ceil/floor/trunc
-# Use $ to denote end of regex so we don't xfail other tests accidentally
-array_api_tests/test_operators_and_elementwise_functions.py::test_ceil
-# TODO: this xfails more than it should ... (e.g. test_floor_divide works)
-array_api_tests/test_operators_and_elementwise_functions.py::test_floor
-array_api_tests/test_operators_and_elementwise_functions.py::test_trunc
-
-# Dask doesn't raise an error for this test
-array_api_tests/test_searching_functions.py::test_nonzero_zerodim_error
-
 # Fails because shape is NaN since we don't materialize it yet
 array_api_tests/test_searching_functions.py::test_nonzero
 array_api_tests/test_set_functions.py::test_unique_all
@@ -80,7 +70,6 @@ array_api_tests/test_set_functions.py::test_unique_values
 
 # Linalg failures (signature failures/missing methods)
 
-
 # fails for ndim > 2
 array_api_tests/test_linalg.py::test_svdvals
 array_api_tests/test_linalg.py::test_cholesky
@@ -139,9 +128,5 @@ array_api_tests/test_has_names.py::test_has_names[array_method-__dlpack_device__
 # Some cases unsupported by dask
 array_api_tests/test_manipulation_functions.py::test_roll
 
-# Dtype doesn't match (output is float32 but should be float64)
-array_api_tests/test_statistical_functions.py::test_prod
-array_api_tests/test_statistical_functions.py::test_sum
-
 # No mT on dask array
 array_api_tests/meta/test_hypothesis_helpers.py::test_symmetric_matrices