code: def testAvgPool(self):
      y = extensions.avg_pool(
          np.ones([5, 320, 480, 3]), [3, 5], [2, 3], "VALID"
      )
      self.assertAllEqual(
          y,
          tf.nn.pool(
              input=tf.ones([5, 320, 480, 3]),
              window_shape=[3, 5],
              pooling_type="AVG",
              padding="VALID",
              strides=[2, 3],
          ),
      )
  
imports: import functools
  import itertools
  from absl import flags
  from absl.testing import parameterized
  from jax import lax
  import numpy
  import tensorflow.compat.v2
  from trax.tf_numpy import extensions
  import trax.tf_numpy.numpy
label_desc: Use image node_id 19 for calling the ExtensionsTest obj's underlying member method code with example usage: obj.testAvgPool() without return types