code: def test_jax_cond(
      *,
      dtype_and_x,
      pred_cond,
      test_flags,
      on_device,
      fn_tree,
      frontend,
      backend_fw,
  ):
      def _test_true_fn(x):
          return x + x
  
      def _test_false_fn(x):
          return x * x
  
      input_dtype, x = dtype_and_x
      helpers.test_frontend_function(
          input_dtypes=input_dtype,
          backend_to_test=backend_fw,
          test_flags=test_flags,
          frontend=frontend,
          fn_tree=fn_tree,
          on_device=on_device,
          pred=pred_cond,
          true_fun=_test_true_fn,
          false_fun=_test_false_fn,
          operand=x[0],
      )
  
file_path: ivy/ivy_tests/test_ivy/test_frontends/test_jax/test_lax/test_control_flow_operators.py
imports: from hypothesis import strategies
  import ivy_tests.test_ivy.helpers
  from ivy_tests.test_ivy.helpers import handle_frontend_test
label_desc: Use image node_id 1 for calling a global function with example usage: test_jax_cond() and returns: unknown, unknown