TST: Test one more case: assigning step through kwarg, and setting it…

… to 1.
dipy · Feb 2, 2016 · bdb2030 · bdb2030
1 parent b77a5a4
commit bdb2030
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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')