from AFQ import dki
import numpy.testing as npt


def test_fit_dki_inputs():
    data_files = ["String in a list"]
    bval_files = "just a string"
    bvec_files = "just another string"
    npt.assert_raises(ValueError, dki.fit_dki, data_files, bval_files,
                      bvec_files)

import tempfile

import numpy.testing as npt
import nibabel.tmpdirs as nbtmp
import dipy.data as dpd

from AFQ import dki


def test_fit_dki_inputs():
    data_files = ["String in a list"]
    bval_files = "just a string"
    bvec_files = "just another string"
    npt.assert_raises(ValueError, dki.fit_dki, data_files, bval_files,
                      bvec_files)

def test_fit_dti():
    fdata, fbval, fbvec = dpd.get_data('small_101D')
    with nbtmp.InTemporaryDirectory() as tmpdir:
        dki.fit_dki(fdata, fbval, fbvec, out_dir=tmpdir)

Add smoke-testing of the DKI fit.
TST: Add smoke-testing of the DKI fit.

+ import tempfile
+ 
+ import numpy.testing as npt
+ import nibabel.tmpdirs as nbtmp
+ import dipy.data as dpd
+ 
  from AFQ import dki
- import numpy.testing as npt
  
  
  def test_fit_dki_inputs():
      data_files = ["String in a list"]
      bval_files = "just a string"
      bvec_files = "just another string"
      npt.assert_raises(ValueError, dki.fit_dki, data_files, bval_files,
                        bvec_files)
  
+ def test_fit_dti():
+     fdata, fbval, fbvec = dpd.get_data('small_101D')
+     with nbtmp.InTemporaryDirectory() as tmpdir:
+         dki.fit_dki(fdata, fbval, fbvec, out_dir=tmpdir)
+ 
Add smoke-testing of the DKI fit.
## Code Before:
from AFQ import dki
import numpy.testing as npt


def test_fit_dki_inputs():
    data_files = ["String in a list"]
    bval_files = "just a string"
    bvec_files = "just another string"
    npt.assert_raises(ValueError, dki.fit_dki, data_files, bval_files,
                      bvec_files)

## Instruction:
Add smoke-testing of the DKI fit.
## Code After:
import tempfile

import numpy.testing as npt
import nibabel.tmpdirs as nbtmp
import dipy.data as dpd

from AFQ import dki


def test_fit_dki_inputs():
    data_files = ["String in a list"]
    bval_files = "just a string"
    bvec_files = "just another string"
    npt.assert_raises(ValueError, dki.fit_dki, data_files, bval_files,
                      bvec_files)

def test_fit_dti():
    fdata, fbval, fbvec = dpd.get_data('small_101D')
    with nbtmp.InTemporaryDirectory() as tmpdir:
        dki.fit_dki(fdata, fbval, fbvec, out_dir=tmpdir)
