compare floats as almost equal

bnavigator · bnavigator · commit 089ca21813b8 · 2022-12-18T21:32:44.000+01:00
diff --git a/control/tests/flatsys_test.py b/control/tests/flatsys_test.py
@@ -693,9 +693,9 @@ def test_response(self, xf, uf, Tf):
 
         # Recompute using response()
         response = traj.response(T, squeeze=False)
-        np.testing.assert_equal(T, response.time)
-        np.testing.assert_equal(u, response.inputs)
-        np.testing.assert_equal(x, response.states)
+        np.testing.assert_array_almost_equal(T, response.time)
+        np.testing.assert_array_almost_equal(u, response.inputs)
+        np.testing.assert_array_almost_equal(x, response.states)
 
     @pytest.mark.parametrize(
         "basis",
@@ -713,7 +713,7 @@ def test_basis_class(self, basis):
             for j in range(basis.N):
                 coefs = np.zeros(basis.N)
                 coefs[j] = 1
-                np.testing.assert_equal(
+                np.testing.assert_array_almost_equal(
                     basis.eval(coefs, timepts),
                     basis.eval_deriv(j, 0, timepts))
         else:
@@ -722,7 +722,7 @@ def test_basis_class(self, basis):
                 for j in range(basis.var_ncoefs(i)):
                     coefs = np.zeros(basis.var_ncoefs(i))
                     coefs[j] = 1
-                    np.testing.assert_equal(
+                    np.testing.assert_array_almost_equal(
                         basis.eval(coefs, timepts, var=i),
                         basis.eval_deriv(j, 0, timepts, var=i))
 
@@ -732,7 +732,7 @@ def test_basis_class(self, basis):
                 for j in range(basis.var_ncoefs(i)):
                     coefs = np.zeros(basis.N)
                     coefs[offset] = 1
-                    np.testing.assert_equal(
+                    np.testing.assert_array_almost_equal(
                         basis.eval(coefs, timepts)[i],
                         basis.eval_deriv(j, 0, timepts, var=i))
                     offset += 1
