code: def test_lazy_local_mask_chunking(
      self,
      block_size: tuple[int, int],
      shape: tuple[int, int],
      window_size: tuple[int | None, int | None],
      offset: int,
  ):
      dense_mask = mask_lib.make_local_attention_mask(
          shape, window_size, offset=offset
      )
      self._compare_masks(
          dense_mask,
          mask_lib.LocalMask(shape, window_size, offset),
          block_size,
      )
  
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 6 for calling the SplashAttentionMaskTest obj's underlying member method code with example usage: obj.test_lazy_local_mask_chunking(block_size, shape, window_size, offset) without return types