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.
