Catch divide-by-zero in step calculation

tonysyu · Sep 2, 2012 · 0d7a1d1 · 0d7a1d1
1 parent 4bbd671
commit 0d7a1d1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/matplotlib/streamplot.py b/lib/matplotlib/streamplot.py
@@ -450,7 +450,10 @@ def _integrate_rk12(x0, y0, dmap, f):
             stotal += ds
 
         # recalculate stepsize based on step error
-        ds = min(maxds, 0.85 * ds * (maxerror/error)**0.5)
+        if error == 0:
+            ds = maxds
+        else:
+            ds = min(maxds, 0.85 * ds * (maxerror/error)**0.5)
 
     return stotal, xf_traj, yf_traj