Skip to content

Commit c04816a

Browse files
committed
Enable tests
1 parent 0683414 commit c04816a

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

tests/test_funcs.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,22 @@
2121
from array_api_extra._lib._testing import xp_assert_close, xp_assert_equal
2222
from array_api_extra._lib._utils._compat import device as get_device
2323
from array_api_extra._lib._utils._typing import Array, Device
24+
from array_api_extra.testing import lazy_xp_function
2425

2526
# some xp backends are untyped
2627
# mypy: disable-error-code=no-untyped-usage
2728

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+
2840

2941
@pytest.mark.skip_xp_backend(Backend.SPARSE, reason="no expand_dims")
3042
class TestAtLeastND:

tests/test_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,13 @@
77
from array_api_extra._lib._utils._compat import device as get_device
88
from array_api_extra._lib._utils._helpers import in1d
99
from array_api_extra._lib._utils._typing import Device
10+
from array_api_extra.testing import lazy_xp_function
1011

1112
# mypy: disable-error-code=no-untyped-usage
1213

14+
# FIXME calls xp.unique_values without size
15+
lazy_xp_function(in1d, jax_jit=False, static_argnames=("assume_unique", "invert", "xp"))
16+
1317

1418
class TestIn1D:
1519
@pytest.mark.skip_xp_backend(Backend.DASK, reason="no argsort")

0 commit comments

Comments
 (0)