code: def test_lazy_full_mask(self, shape: tuple[int, int]):
      lazy_full = mask_lib.FullMask(shape)
      dense = np.ones(shape, dtype=np.bool_)
  
      self._compare_masks(dense, lazy_full, (256, 256))
  
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 11 for calling the SplashAttentionMaskTest obj's underlying member method code with example usage: obj.test_lazy_full_mask(shape) without return types