code: def testAnalyticKernelComposeAutomatic(
      self, store_on_device, batch_size
  ):
      test_utils.stub_out_pmap(batching, 2)
      self._test_analytic_kernel_composition(
          partial(
              batching.batch,
              batch_size=batch_size,
              store_on_device=store_on_device,
          )
      )
  
imports: from functools import partial
  from absl.testing import absltest
  from jax import jit
  from jax import config
  from jax import random
  import jax.numpy
  from jax.tree_util import tree_map
  import neural_tangents
  from neural_tangents import stax
  from neural_tangents._src import batching
  from neural_tangents._src.empirical import _DEFAULT_TESTING_NTK_IMPLEMENTATION
  from tests import test_utils
label_desc: Use image node_id 9 for calling the BatchTest obj's underlying member method code with example usage: obj.testAnalyticKernelComposeAutomatic(store_on_device, batch_size) without return types