diff --git a/tests/test_array_namespace.py b/tests/test_array_namespace.py index 311efc37..7f196387 100644 --- a/tests/test_array_namespace.py +++ b/tests/test_array_namespace.py @@ -24,7 +24,8 @@ def test_array_namespace(request, library, api_version, use_compat): pytest.raises(ValueError, lambda: array_namespace(array, use_compat=use_compat)) return if (library == "sparse" and api_version in ("2023.12", "2024.12")) or ( - library == "jax.numpy" and api_version in ("2021.12", "2022.12", "2023.12") + library == "jax.numpy" and + api_version in ("2021.12", "2022.12", "2023.12", "2024.12") ): xfail(request, "Unsupported API version") diff --git a/tests/test_common.py b/tests/test_common.py index 85ed032e..1ebf6e7a 100644 --- a/tests/test_common.py +++ b/tests/test_common.py @@ -203,7 +203,7 @@ def test_device_to_device(library, request): devices = xp.__array_namespace_info__().devices() # Default device - x = xp.asarray([1, 2, 3]) + x = xp.asarray([1, 2, 3], dtype=xp.int32) dev = device(x) for dev in devices: