old_contents: import os.path as op
  
  import numpy as np
  import numpy.testing as npt
  
  import nibabel as nib
  import nibabel.tmpdirs as nbtmp
  
  import dipy.data as dpd
  from dipy.reconst.shm import calculate_max_order
  
  from AFQ import csd
  
  
  def test_fit_csd():
      fdata, fbval, fbvec = dpd.get_data()
      with nbtmp.InTemporaryDirectory() as tmpdir:
          # Convert from npy to txt:
          bvals = np.load(fbval)
          bvecs = np.load(fbvec)
          np.savetxt(op.join(tmpdir, 'bvals.txt'), bvals)
          np.savetxt(op.join(tmpdir, 'bvecs.txt'), bvecs)
          for sh_order in [4, 8]:
              fname = csd.fit_csd(fdata, op.join(tmpdir, 'bvals.txt'),
                                  op.join(tmpdir, 'bvecs.txt'),
                                  out_dir=tmpdir, sh_order=sh_order)
              npt.assert_(op.exists(fname))
              sh_coeffs_img = nib.load(fname)
              npt.assert_equal(sh_order,
                               calculate_max_order(sh_coeffs_img.shape[-1]))
  
new_contents: import os.path as op
  
  import numpy as np
  import numpy.testing as npt
  
  import nibabel as nib
  import nibabel.tmpdirs as nbtmp
  
  import dipy.data as dpd
  from dipy.reconst.shm import calculate_max_order
  
  from AFQ import csd
  
  
  def test_fit_csd():
      fdata, fbval, fbvec = dpd.get_data('small_64D')
      with nbtmp.InTemporaryDirectory() as tmpdir:
          # Convert from npy to txt:
          bvals = np.load(fbval)
          bvecs = np.load(fbvec)
          np.savetxt(op.join(tmpdir, 'bvals.txt'), bvals)
          np.savetxt(op.join(tmpdir, 'bvecs.txt'), bvecs)
          for sh_order in [4, 6]:
              fname = csd.fit_csd(fdata, op.join(tmpdir, 'bvals.txt'),
                                  op.join(tmpdir, 'bvecs.txt'),
                                  out_dir=tmpdir, sh_order=sh_order)
              npt.assert_(op.exists(fname))
              sh_coeffs_img = nib.load(fname)
              npt.assert_equal(sh_order,
                               calculate_max_order(sh_coeffs_img.shape[-1]))
  
subject: Replace the test data set with this one.