code: def test_max_pool_unbatched(self):
      x = np.arange(6, dtype=jnp.float32).reshape([6, 1])
      leading_dims = (2, 3)
      x = np.broadcast_to(x, leading_dims + (10, 6, 2))
  
      window_shape = [2, 2, 1]
      result = pool.max_pool(
          x,
          window_shape=window_shape,
          strides=window_shape,
          padding="VALID",
      )
  
      ground_truth = np.asarray([1.0, 3.0, 5.0]).reshape([3, 1])
      ground_truth = np.broadcast_to(
          ground_truth, leading_dims + (5, 3, 2)
      )
  
      np.testing.assert_equal(result, ground_truth)
  
imports: import functools
  from absl.testing import absltest
  from haiku._src import pool
  from haiku._src import test_utils
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling the MaxPoolTest obj's underlying member method code with example usage: obj.test_max_pool_unbatched() without return types