code: def test_compute_mask_indices(self):
      batch_size = 4
      sequence_length = 60
      mask_prob = 0.5
      mask_length = 1
  
      mask = _compute_mask_indices(
          (batch_size, sequence_length), mask_prob, mask_length
      )
  
      self.assertListEqual(
          mask.sum(axis=-1).tolist(),
          [mask_prob * sequence_length for _ in range(batch_size)],
      )
  
imports: import inspect
  import math
  import multiprocessing
  import traceback
  import unittest
  import numpy
  from datasets import load_dataset
  from transformers import Wav2Vec2Config, is_flax_available
  from transformers.testing_utils import CaptureLogger, is_flaky, is_librosa_available, is_pt_flax_cross_test, is_pyctcdecode_available, require_flax, require_librosa, require_pyctcdecode, require_soundfile, run_test_in_subprocess, slow
  from ...test_modeling_flax_common import FlaxModelTesterMixin, floats_tensor, random_attention_mask
label_desc: Use image node_id 1 for calling the FlaxWav2Vec2UtilsTest obj's underlying member method code with example usage: obj.test_compute_mask_indices() without return types