fix bug with array==0! problem test in matlab_test.py

gonmolina · gonmolina · commit 8b829ccfcc05 · 2017-03-21T00:13:08.000-03:00
diff --git a/control/rlocus.py b/control/rlocus.py
@@ -225,7 +225,7 @@ def _k_max(num, den, real_break_points, k_break_points):
     else:
         farthest_points = 2 * np.max(np.abs(important_points))
         kmax_asymp = [-den(farthest_points) / num(farthest_points)]
-    if kmax_asymp == 0:
+    if np.max(kmax_asymp) == 0:
         kmax_asymp = [den.coeffs[0] / num.coeffs[0] * 3]
     kmax = np.max(np.concatenate((np.real(kmax_asymp), k_break_points), axis=0))
     return kmax
