code: def test_default_weights_batch():
      file_path = get_fnames("evac_test_data")
      input_arr = np.load(file_path)["input"]
      output_arr = np.load(file_path)["output"]
      input_arr = [T1 for T1 in input_arr]
  
      evac_model = EVACPlus()
      fake_affine = np.array([np.eye(4), np.eye(4)])
      fake_voxsize = np.ones((2, 3))
      results_arr = evac_model.predict(
          input_arr,
          fake_affine,
          fake_voxsize,
          batch_size=2,
          return_prob=True,
      )
      npt.assert_almost_equal(results_arr, output_arr, decimal=4)
  
file_path: dipy/dipy/nn/tests/test_evac.py
imports: import pytest
  import numpy
  import numpy.testing
  from dipy.data import get_fnames
  from dipy.utils.optpkg import optional_package
label_desc: Use image node_id 2 for calling a global function with example usage: test_default_weights_batch() without return types