|
21 | 21 | from array_api_extra._lib._testing import xp_assert_close, xp_assert_equal |
22 | 22 | from array_api_extra._lib._utils._compat import device as get_device |
23 | 23 | from array_api_extra._lib._utils._typing import Array, Device |
| 24 | +from array_api_extra.testing import lazy_xp_function |
24 | 25 |
|
25 | 26 | # some xp backends are untyped |
26 | 27 | # mypy: disable-error-code=no-untyped-usage |
27 | 28 |
|
| 29 | +lazy_xp_function(atleast_nd, static_argnames=("ndim", "xp")) |
| 30 | +lazy_xp_function(cov, static_argnames="xp") |
| 31 | +lazy_xp_function(create_diagonal, static_argnames=("offset", "xp")) |
| 32 | +lazy_xp_function(expand_dims, static_argnames=("axis", "xp")) |
| 33 | +lazy_xp_function(kron, static_argnames="xp") |
| 34 | +lazy_xp_function(nunique, static_argnames="xp") |
| 35 | +lazy_xp_function(pad, static_argnames=("pad_width", "mode", "constant_values", "xp")) |
| 36 | +# FIXME calls in1d which calls xp.unique_values without size |
| 37 | +lazy_xp_function(setdiff1d, jax_jit=False, static_argnames=("assume_unique", "xp")) |
| 38 | +lazy_xp_function(sinc, static_argnames="xp") |
| 39 | + |
28 | 40 |
|
29 | 41 | @pytest.mark.skip_xp_backend(Backend.SPARSE, reason="no expand_dims") |
30 | 42 | class TestAtLeastND: |
|
0 commit comments