code: def testAutomatic(
      self,
      train_size,
      test_size,
      input_shape,
      network,
      kernel_type,
      batch_size,
  ):
      data_other, data_self, kernel_fn = self._get_data_and_kernel_fn(
          input_shape, kernel_type, network, test_size, train_size
      )
  
      kernel_batched = batching.batch(kernel_fn, batch_size=batch_size)
      _test_kernel_against_batched(
          self, kernel_fn, kernel_batched, data_self, data_other
      )
  
      kernel_batched = batching.batch(
          kernel_fn, batch_size=batch_size, store_on_device=False
      )
      _test_kernel_against_batched(
          self, kernel_fn, kernel_batched, data_self, data_other
      )
  
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 5 for calling the BatchTest obj's underlying member method code with example usage: obj.testAutomatic(train_size, test_size, input_shape, network, kernel_type, batch_size) without return types