Use assert_array_equal to compare results in scalar timeresp test

roryyorke · roryyorke · commit a05b9d50f830 · 2025-02-01T18:59:00.000+02:00
diff --git a/control/tests/flatsys_test.py b/control/tests/flatsys_test.py
@@ -470,9 +470,9 @@ def terminal_cost(x, u):
         r1 = traj1.response(teval)
         r2 = traj2.response(teval)
 
-        assert np.max(abs(r1.x-r2.x)) == 0
-        assert np.max(abs(r1.u-r2.u)) == 0
-        assert np.max(abs(r1.y-r2.y)) == 0
+        np.testing.assert_array_equal(r1.x, r2.x)
+        np.testing.assert_array_equal(r1.y, r2.y)
+        np.testing.assert_array_equal(r1.u, r2.u)
 
 
     def test_bezier_basis(self):
