code: def testSparseSplit(self, compile):
      f = jit(split) if compile else split
      rng = jtu.rand_default(self.rng())
      M = make_sparse_array(rng, (10,), jnp.float32)
      M2, M3 = f(M)
  
      jaxpr = make_jaxpr(f)(M).jaxpr
      core.check_jaxpr(jaxpr)
  
      for MM in M2, M3:
          self.assertEqual(M.dtype, MM.dtype)
          self.assertEqual(M.index_dtype, MM.index_dtype)
          self.assertArraysEqual(M.data, MM.data)
          self.assertArraysEqual(M.indices, MM.indices)
  
imports: from absl.testing import absltest
  import math
  import unittest
  import numpy
  import jax.numpy
  from jax import jit, lax, make_jaxpr
  from jax import config
  from jax.interpreters import mlir
  from jax.interpreters import xla
  from jax._src import core
  from jax._src import dtypes
  from jax._src import test_util
  from jax._src import xla_bridge
  from jax._src.lib.mlir import ir
  from jax._src.lib import xla_client
label_desc: Use image node_id 2 for calling the CustomObjectTest obj's underlying member method code with example usage: obj.testSparseSplit(compile) without return types