add test for matlab.stepinfo

bnavigator · bnavigator · commit da1c517ed4f4 · 2020-08-01T15:46:45.000+02:00
diff --git a/control/tests/matlab_test.py b/control/tests/matlab_test.py
@@ -16,7 +16,7 @@
 from control.matlab import ss, ss2tf, ssdata, tf, tf2ss, tfdata, rss, drss, frd
 from control.matlab import parallel, series, feedback
 from control.matlab import pole, zero, damp
-from control.matlab import step, impulse, initial, lsim
+from control.matlab import step, stepinfo, impulse, initial, lsim
 from control.matlab import margin, dcgain
 from control.matlab import linspace, logspace
 from control.matlab import bode, rlocus, nyquist, nichols, ngrid, pzmap
@@ -215,6 +215,12 @@ def testStep_mimo(self, mimo):
         np.testing.assert_array_almost_equal(y_00, youttrue, decimal=4)
         np.testing.assert_array_almost_equal(y_11, youttrue, decimal=4)
 
+    def testStepinfo(self, siso):
+        """Test the stepinfo function"""
+        infodict = stepinfo(siso.ss1)
+        assert isinstance(infodict, dict)
+        assert len(infodict) == 9
+
     def testImpulse(self, siso):
         t = np.linspace(0, 1, 10)
         # test transfer function
@@ -344,31 +350,29 @@ def testMargin(self, siso):
 
     def testDcgain(self, siso):
         """Test dcgain for SISO system"""
-        #Create different forms of a SISO system
-        A, B, C, D = siso.ss1.A, siso.ss1.B, siso.ss1.C, \
-                     siso.ss1.D
+        # Create different forms of a SISO system using scipy.signal
+        A, B, C, D = siso.ss1.A, siso.ss1.B, siso.ss1.C, siso.ss1.D
         Z, P, k = sp.signal.ss2zpk(A, B, C, D)
         num, den = sp.signal.ss2tf(A, B, C, D)
         sys_ss = siso.ss1
 
-        #Compute the gain with ``dcgain``
+        # Compute the gain with ``dcgain``
         gain_abcd = dcgain(A, B, C, D)
         gain_zpk = dcgain(Z, P, k)
         gain_numden = dcgain(np.squeeze(num), den)
         gain_sys_ss = dcgain(sys_ss)
         # print('\ngain_abcd:', gain_abcd, 'gain_zpk:', gain_zpk)
         # print('gain_numden:', gain_numden, 'gain_sys_ss:', gain_sys_ss)
 
-        #Compute the gain with a long simulation
+        # Compute the gain with a long simulation
         t = linspace(0, 1000, 1000)
         y, _t = step(sys_ss, t)
         gain_sim = y[-1]
         # print('gain_sim:', gain_sim)
 
-        #All gain values must be approximately equal to the known gain
+        # All gain values must be approximately equal to the known gain
         np.testing.assert_array_almost_equal(
-            [gain_abcd, gain_zpk, gain_numden, gain_sys_ss,
-             gain_sim],
+            [gain_abcd, gain_zpk, gain_numden, gain_sys_ss, gain_sim],
             [59, 59, 59, 59, 59])
 
     def testDcgain_mimo(self, mimo):
