class_docstr: Check the construction of MaskInfo from Mask.
class_docstr_tok: ["Check","the","construction","of","MaskInfo","from","Mask","."]
code: def _process_mask(self, *args, **kwargs):
      mask_info, mask_function = mask_info_lib.process_mask(
          *args, **kwargs
      )
      mask_info_dkv, dkv_mask_function = mask_info_lib.process_mask_dkv(
          *args, **kwargs
      )
      self.assertEqual(mask_function, dkv_mask_function)
      return mask_info, mask_info_dkv, mask_function
  
imports: from __future__ import annotations
  from typing import List
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax.experimental.pallas.ops.tpu.splash_attention import splash_attention_mask
  from jax.experimental.pallas.ops.tpu.splash_attention import splash_attention_mask_info
  from jax._src import test_util
  import numpy
label_desc: Use image node_id 2 for calling the SplashAttentionMaskInfoTest obj's underlying member method code with example usage: obj._process_mask() and returns: mask_info,  mask_info_dkv,  mask_function