From 34d1c598b381a6fbb958e16e363213636279aca7 Mon Sep 17 00:00:00 2001 From: Sasha Lopoukhine Date: Sat, 17 Aug 2024 19:31:36 +0100 Subject: [PATCH] explicit tuple equality --- tests/backend/test_jax_executable.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/backend/test_jax_executable.py b/tests/backend/test_jax_executable.py index 13e8503ac8..1499ebd34c 100644 --- a/tests/backend/test_jax_executable.py +++ b/tests/backend/test_jax_executable.py @@ -39,14 +39,14 @@ def test_abs(): @executable def abs_tuple(a: jax.Array) -> tuple[jax.Array]: ... - assert abs_tuple(array(-2, dtype=jax.numpy.int32))[0] == array( - 2, dtype=jax.numpy.int32 + assert abs_tuple(array(-2, dtype=jax.numpy.int32)) == ( + array(2, dtype=jax.numpy.int32), ) - assert abs_tuple(array(0, dtype=jax.numpy.int32))[0] == array( - 0, dtype=jax.numpy.int32 + assert abs_tuple(array(0, dtype=jax.numpy.int32)) == ( + array(0, dtype=jax.numpy.int32), ) - assert abs_tuple(array(2, dtype=jax.numpy.int32))[0] == array( - 2, dtype=jax.numpy.int32 + assert abs_tuple(array(2, dtype=jax.numpy.int32)) == ( + array(2, dtype=jax.numpy.int32), ) @executable