format test_func.py

Author: Enderdead

tests/test_funcs.py

@@ -34,7 +34,8 @@
 )
 from array_api_extra._lib._backends import NUMPY_VERSION, Backend
 from array_api_extra._lib._testing import xp_assert_close, xp_assert_equal
-from array_api_extra._lib._utils._compat import device as get_device, is_jax_namespace
+from array_api_extra._lib._utils._compat import device as get_device
+from array_api_extra._lib._utils._compat import is_jax_namespace
 from array_api_extra._lib._utils._helpers import eager_shape, ndindex
 from array_api_extra._lib._utils._typing import Array, Device
 from array_api_extra.testing import lazy_xp_function
@@ -1270,7 +1271,7 @@ def test_shapes(
         x1 = xp.zeros(shape1)
         x2 = xp.zeros(shape2)
 
-        if is_jax_namespace(xp) and assume_unique and shape1!=(1,):
+        if is_jax_namespace(xp) and assume_unique and shape1 != (1,):
             pytest.xfail(reason="jax#32335 fixed with jax>=0.8.0")
 
         actual = setdiff1d(x1, x2, assume_unique=assume_unique)