code: def test_source_locations(self):
      def f(x, y):
          s = jnp.sin(x)  # sin
          return jnp.sin(s) + jnp.cos(y)  # sin, cos, add
  
      hist = jaxpr_util.source_locations(make_jaxpr(f)(1.0, 1.0).jaxpr)
      self.assertEqual(sum(hist.values()), 4)
  
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 4 for calling the JaxprStatsTest obj's underlying member method code with example usage: obj.test_source_locations() and returns: unknown