Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/array_api_extra/testing.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

import pytest

from array_api_extra._lib._utils._compat import is_dask_namespace, is_jax_namespace
from ._lib._utils._compat import is_dask_namespace, is_jax_namespace

__all__ = ["lazy_xp_function", "patch_lazy_xp_functions"]

Expand Down
9 changes: 9 additions & 0 deletions vendor_tests/test_vendor.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,15 @@ def test_vendor_extra():
assert_array_equal(y, x)


def test_vendor_extra_testing():
from .array_api_extra.testing import lazy_xp_function

def f(x):
return x

lazy_xp_function(f)


def test_vendor_extra_uses_vendor_compat():
from ._array_api_compat_vendor import array_namespace as n1
from .array_api_extra._lib._utils._compat import array_namespace as n2
Expand Down
Loading