address review: allclose broadcasts

bnavigator · bnavigator · commit 66b381f75c46 · 2021-03-25T13:23:58.000+01:00
diff --git a/control/tests/timeresp_test.py b/control/tests/timeresp_test.py
@@ -883,8 +883,7 @@ def test_time_vector(self, tsystem, fun, squeeze, matarrayout):
             # tout should always match t, which has shape (n, )
             np.testing.assert_allclose(tout, tsystem.t)
         elif fun == forced_response and sys.dt in [None, True]:
-            np.testing.assert_allclose(
-                np.diff(tout), np.full_like(tout[:-1], 1.))
+            np.testing.assert_allclose(np.diff(tout), 1.)
 
         if squeeze is False or not sys.issiso():
             assert yout.shape[0] == sys.noutputs
diff --git a/control/timeresp.py b/control/timeresp.py
@@ -356,7 +356,7 @@ def forced_response(sys, T=None, U=0., X0=0., transpose=False,
 
     # equally spaced also implies strictly monotonic increase
     dt = T[1] - T[0]
-    if not np.allclose(np.diff(T), np.full_like(T[:-1], dt)):
+    if not np.allclose(np.diff(T), dt):
         raise ValueError("Parameter ``T``: time values must be "
                          "equally spaced.")
     n_steps = T.shape[0]            # number of simulation steps
@@ -365,7 +365,6 @@ def forced_response(sys, T=None, U=0., X0=0., transpose=False,
     X0 = _check_convert_array(X0, [(n_states,), (n_states, 1)],
                               'Parameter ``X0``: ', squeeze=True)
 
-
     xout = np.zeros((n_states, n_steps))
     xout[:, 0] = X0
     yout = np.zeros((n_outputs, n_steps))
