code: def test_pprof_equation_profile(self):
      def f(x, y):
          s = jit(jnp.sin)(x)
          return jnp.sin(s) + jnp.cos(y)
  
      profile_gz = jaxpr_util.pprof_equation_profile(
          make_jaxpr(f)(1.0, 1.0).jaxpr
      )
      profile_proto = gzip.decompress(profile_gz)
      json_str = xla_client._xla.pprof_profile_to_json(profile_proto)
      profile = json.loads(json_str)
      self.assertSetEqual(
          {
              "sampleType",
              "sample",
              "stringTable",
              "location",
              "function",
          },
          set(profile.keys()),
      )
  
file_path: jax/tests/jaxpr_util_test.py
imports: import os
  import gzip
  import json
  from absl.testing import absltest
  import jax
  from jax import jit, make_jaxpr, numpy
  from jax._src import config
  from jax._src import jaxpr_util
  from jax._src import test_util
  from jax._src.lib import xla_client
label_desc: Use image node_id 7 for calling the JaxprStatsTest obj's underlying member method code with example usage: obj.test_pprof_equation_profile() and returns: unknown