def METHOD_NAME(self):
    input_shape = (None, 4, 4, 1)
    pool_size = (2, 2)
    strides = (2, 2)
    output_shape = operation_utils.compute_pooling_output_shape(
        input_shape, pool_size, strides
    )
    expected_output_shape = (None, 2, 2, 1)
    self.assertEqual(output_shape, expected_output_shape)