Fix test for gear and multirotor - part2

Author: jguarato

ross/gear_element.py

@@ -913,7 +913,7 @@ class Mesh:
     ... )
     >>> mesh = Mesh(driving, driven)
     >>> mesh.stiffness # doctest : +ELLIPSIS
-    429787128.434...
+    429787095.100...
     """
 
     def __init__(

ross/multi_rotor.py

@@ -108,7 +108,7 @@ class MultiRotor(Rotor):
     ... )
     >>> modal = multi_rotor.run_modal(speed=0)
     >>> modal.wd[0] # doctest: +ELLIPSIS
-    74.163...
+    74.160...
     """
 
     @check_units

ross/tests/test_gear_element.py

@@ -1,7 +1,7 @@
 import numpy as np
 import pytest
 from copy import deepcopy
-from numpy.testing import assert_almost_equal
+from numpy.testing import assert_allclose
 
 from ross.units import Q_
 from ross.materials import Material, steel
@@ -38,7 +38,7 @@ def test_mass_matrix_gear(gear):
                    [     0.,      0.,      0.,      0.,      0., 113.9]])
     # fmt: on
 
-    assert_almost_equal(gear.M(), Mg, decimal=5)
+    assert_allclose(gear.M(), Mg, rtol=1e-6, atol=1e-5)
 
 
 def test_gyroscopic_matrix_gear(gear):
@@ -51,7 +51,7 @@ def test_gyroscopic_matrix_gear(gear):
                    [0., 0., 0.,     0.,    0., 0.]])
     # fmt: on
 
-    assert_almost_equal(gear.G(), Gg, decimal=5)
+    assert_allclose(gear.G(), Gg, rtol=1e-6, atol=1e-5)
 
 
 @pytest.fixture
@@ -114,7 +114,7 @@ def test_mass_matrix_gear_tvms(gear_tvms):
                     [     0.,      0.,      0.,      0.,      0., 0.00360]])
     # fmt: on
 
-    assert_almost_equal(gear_tvms.M(), Mgt, decimal=5)
+    assert_allclose(gear_tvms.M(), Mgt, rtol=1e-6, atol=1e-5)
 
 
 def test_gyroscopic_matrix_gear_tvms(gear_tvms):
@@ -127,7 +127,7 @@ def test_gyroscopic_matrix_gear_tvms(gear_tvms):
                     [0., 0., 0.,       0.,      0., 0.]])
     # fmt: on
 
-    assert_almost_equal(gear_tvms.G(), Ggt, decimal=5)
+    assert_allclose(gear_tvms.G(), Ggt, rtol=1e-6, atol=1e-5)
 
 
 @pytest.fixture

ross/tests/test_multi_rotor.py

@@ -1,7 +1,7 @@
 import numpy as np
 import pytest
 from copy import deepcopy
-from numpy.testing import assert_almost_equal
+from numpy.testing import assert_allclose
 
 import ross as rs
 
@@ -125,8 +125,10 @@ def multi_rotor():
 
 
 def test_mesh(multi_rotor):
-    assert multi_rotor.mesh.contact_ratio == 1.6377334309511222
-    assert multi_rotor.mesh.stiffness == 1937234387.18946
+    assert_allclose(
+        multi_rotor.mesh.contact_ratio, 1.6377334309511222, rtol=1e-6, atol=1e-5
+    )
+    assert_allclose(multi_rotor.mesh.stiffness, 1937234387.18946, rtol=1e-6, atol=1e-5)
 
 
 def test_coupling_matrix_gear(multi_rotor):
@@ -303,4 +305,6 @@ def test_coupling_matrix_gear(multi_rotor):
         ]
     )
 
-    assert_almost_equal(multi_rotor.coupling_matrix(), coupling_matrix, decimal=5)
+    assert_allclose(
+        multi_rotor.coupling_matrix(), coupling_matrix, rtol=1e-6, atol=1e-5
+    )