code: def test_cond_metadata(self):
      def true_fun(x):
          return jnp.sin(x)
  
      def false_fun(x):
          return jnp.cos(x)
  
      def f(which, x):
          return jax.lax.cond(which, x, true_fun, x, false_fun)
  
      hlo = module_to_string(jax.jit(f).lower(True, 1.0).compiler_ir())
      self.assertRegex(hlo, r'loc\(".*cond\[linear=\(False, False\)\]"')
      self.assertRegex(hlo, r'loc\(".*cond/branch_0_fun/cos"')
      self.assertRegex(hlo, r'loc\(".*cond/branch_1_fun/sin"')
  
file_path: jax/tests/metadata_test.py
imports: import io
  import unittest
  from absl.testing import absltest
  from jax._src import test_util
  import jax
  from jax._src import config
  from jax._src.lib.mlir import ir
  from jax import numpy
  from jax import config
label_desc: Use image node_id 4 for calling the MetadataTest obj's underlying member method code with example usage: obj.test_cond_metadata() and returns: jnp, jnp, jax