added small checks for return_x argument in step

murrayrm · murrayrm · commit 80bd3d572bc8 · 2016-05-30T11:39:56.000-07:00
diff --git a/control/tests/test_control_matlab.py b/control/tests/test_control_matlab.py
@@ -145,6 +145,9 @@ def test_step(self):
         t, y = step(sys, T, X0)
         plot(t, y)
 
+        # Test output of state vector
+        t, y, x = step(sys, return_x=True)
+
         #Test MIMO system
         A, B, C, D = self.make_MIMO_mats()
         sys = ss(A, B, C, D)
diff --git a/control/tests/timeresp_test.py b/control/tests/timeresp_test.py
@@ -66,6 +66,10 @@ def test_step_response(self):
         np.testing.assert_array_almost_equal(yout, youttrue, decimal=4)
         np.testing.assert_array_almost_equal(tout, t)
 
+        tout, yout, xout = step_response(sys, T=t, X0=0, return_x=True)
+        np.testing.assert_array_almost_equal(yout, youttrue, decimal=4)
+        np.testing.assert_array_almost_equal(tout, t)
+
         # Test MIMO system, which contains ``siso_ss1`` twice
         sys = self.mimo_ss1
         _t, y_00 = step_response(sys, T=t, input=0, output=0)
