class_docstr: Tests for modules in masked_layers.py.
class_docstr_tok: ["Tests","for","modules","in","masked_layers.py","."]
docstr: Masked pool with 'SAME' padding is not supported.
docstr_tok: ["Masked","pool","with","'SAME","'","padding","is","not","supported","."]
code: def test_masked_same_pool_raises(
      self, pool_fn, window_shape, strides, padding
  ):
      """Masked pool with 'SAME' padding is not supported."""
      _, masked_pool_fn = self.POOL_FN_DICT[pool_fn]
  
      with self.assertRaises(NotImplementedError):
          masked_pool_fn(
              self.INPUTS_PADDED,
              window_shape,
              strides,
              padding=padding,
              spatial_shape=self.SPATIAL_SHAPE,
          )
  
imports: import dataclasses
  from typing import Any, Callable, Dict, Sequence, Tuple, Type, Union
  from absl.testing import absltest
  from absl.testing import parameterized
  import flax.linen
  from jax import random
  import jax.numpy
  import numpy
  from scenic.model_lib.layers import masked_layers
label_desc: Use image node_id 5 for calling the MaskedLayersTest obj's underlying member method code with example usage: obj.test_masked_same_pool_raises(pool_fn, window_shape, strides, padding) without return types