code: def get_brain_models():
      """
      Generates a set of practice BrainModelAxis axes
  
      Yields
      ------
      BrainModelAxis axis
      """
      mask = np.zeros(vol_shape)
      mask[0, 1, 2] = 1
      mask[0, 4, 2] = True
      mask[0, 4, 0] = True
      yield axes.BrainModelAxis.from_mask(
          mask, "ThalamusRight", rand_affine
      )
      mask[0, 0, 0] = True
      yield axes.BrainModelAxis.from_mask(mask, affine=rand_affine)
  
      yield axes.BrainModelAxis.from_surface(
          [0, 5, 10], 15, "CortexLeft"
      )
      yield axes.BrainModelAxis.from_surface([0, 5, 10, 13], 15)
  
      surface_mask = np.zeros(15, dtype="bool")
      surface_mask[[2, 9, 14]] = True
      yield axes.BrainModelAxis.from_mask(
          surface_mask, name="CortexRight"
      )
  
file_path: nibabel/nibabel/cifti2/tests/test_axes.py
imports: from copy import deepcopy
  import numpy
  import pytest
  import nibabel.cifti2.cifti2_axes
  from .test_cifti2io_axes import check_rewrite
label_desc: Use image node_id 1 for calling a global function with example usage: get_brain_models() without return types