fix PEP8 compliance style and preserve pole type

vincentchoqueuse · vincentchoqueuse · commit 140ca08fc7ad · 2021-08-18T08:43:42.000+02:00
diff --git a/control/lti.py b/control/lti.py
@@ -159,8 +159,7 @@ def damp(self):
         poles = self.pole()
 
         if isdtime(self, strict=True):
-            poles = poles.astype(complex)
-            splane_poles = np.log(poles)/self.dt
+            splane_poles = np.log(poles.astype(complex))/self.dt
         else:
             splane_poles = poles
         wn = absolute(splane_poles)
diff --git a/control/tests/lti_test.py b/control/tests/lti_test.py
@@ -70,15 +70,14 @@ def test_damp(self):
         np.testing.assert_almost_equal(sys_dt.damp(), expected_dt)
         np.testing.assert_almost_equal(damp(sys_dt), expected_dt)
 
-        #also check that for a discrete system with a negative real pole the damp function can extract wn and theta.
+        #also check that for a discrete system with a negative real pole the damp function can extract wn and zeta.
         p2_zplane = -0.2
-        sys_dt2 = tf(1,[1,-p2_zplane],dt)
-        wn2, zeta2, _ = sys_dt2.damp()
-        p2 = -wn2 * zeta2 + 1j * wn2 * np.sqrt(1 - zeta2**2)
-        p2_zplane = np.exp(p2*dt)
-        np.testing.assert_almost_equal(sys_dt2.pole(),p2_zplane)
+        sys_dt2 = tf(1, [1, -p2_zplane], dt)
+        wn2, zeta2, p2 = sys_dt2.damp()
+        p2_splane = -wn2 * zeta2 + 1j * wn2 * np.sqrt(1 - zeta2**2)
+        p2_zplane = np.exp(p2_splane * dt)
+        np.testing.assert_almost_equal(p2, p2_zplane)
         
-
     def test_dcgain(self):
         sys = tf(84, [1, 2])
         np.testing.assert_allclose(sys.dcgain(), 42)
