From bdb20308cc447ec173f76e225ea15e249f052c1b Mon Sep 17 00:00:00 2001
From: arokem <arokem@gmail.com>
Date: Tue, 2 Feb 2016 15:14:47 -0800
Subject: [PATCH] TST: Test one more case: assigning step through kwarg, and
 setting it to 1.

---
 dipy/reconst/tests/test_dti.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/dipy/reconst/tests/test_dti.py b/dipy/reconst/tests/test_dti.py
index e6ad3537ae..2a6c329ae2 100644
--- a/dipy/reconst/tests/test_dti.py
+++ b/dipy/reconst/tests/test_dti.py
@@ -685,7 +685,9 @@ def test_predict():
     S0 = 1
     p = dtif.predict(gtab, S0)
     assert_equal(p.shape, data.shape)
-
+    # Assign the step through kwarg:
+    p = dtif.predict(gtab, S0, step=1)
+    assert_equal(p.shape, data.shape)
 
 def test_eig_from_lo_tri():
     psphere = get_sphere('symmetric362')