Merge pull request #1277 from gsabinoo/feat/ruff-format

Add `ruff format --check` to CI

Author: raphaeltimbo

.github/workflows/test.yml

@@ -36,9 +36,10 @@ jobs:
         run: |
           pip install pytest
           pytest ross --cov=ross
-      - name: Run Ruff 
+      - name: Run Ruff
         run: |
           ruff check ross
+          ruff format --check ross
         if: success()
       - name: Coverage and Deployment
         run: codecov

ross/bearing_seal_element.py

@@ -1487,7 +1487,6 @@ def __init__(
     ):
         self.seal_leakage = seal_leakage
 
-
         super().__init__(
             n=n,
             frequency=frequency,

ross/bearings/magnetic/controllers.py

@@ -85,7 +85,7 @@ def plot_frequency_response(*systems, **kwargs):
     ]
 
     if legends is None:
-        legends = [f"System {i+1}" for i in range(len(systems))]
+        legends = [f"System {i + 1}" for i in range(len(systems))]
     elif len(legends) != len(systems):
         raise ValueError("O número de legendas deve ser igual ao número de sistemas.")
 

ross/bearings/plain_journal.py

@@ -287,8 +287,8 @@ def __init__(
         a, b = self._get_interp_coeffs(
             T_muI, T_muF, mu_I, mu_F
         )  # Interpolation coefficients
-        self.interpolate = lambda reference: a * (
-            reference**b
+        self.interpolate = lambda reference: (
+            a * (reference**b)
         )  # Interpolation function
         self.reference_viscosity = self.interpolate(self.reference_temperature)
 

ross/rotor_assembly.py

@@ -946,11 +946,11 @@ def run_critical_speed(self, speed_range=None, num_modes=12, rtol=0.005):
         wd = np.zeros_like(_wd)
 
         for i in range(len(wn)):
-            wn_func = lambda s: (s - self.run_modal(s, num_modes).wn[i])
+            wn_func = lambda s: s - self.run_modal(s, num_modes).wn[i]
             wn[i] = newton(func=wn_func, x0=_wn[i], rtol=rtol)
 
         for i in range(len(wd)):
-            wd_func = lambda s: (s - self.run_modal(s, num_modes).wd[i])
+            wd_func = lambda s: s - self.run_modal(s, num_modes).wd[i]
             wd[i] = newton(func=wd_func, x0=_wd[i], rtol=rtol)
 
         log_dec = np.zeros_like(wn)
@@ -2577,26 +2577,32 @@ def integrate_system(self, speed, F, t, **kwargs):
         add_to_RHS = kwargs.get("add_to_RHS")
 
         if add_to_RHS is None:
-            forces = lambda step, **curr_state: F[step, :] + reduction[1](
-                magnetic_force(
-                    step,
-                    curr_state.get("dt"),
-                    reduction[2](curr_state.get("y")),
+            forces = lambda step, **curr_state: (
+                F[step, :]
+                + reduction[1](
+                    magnetic_force(
+                        step,
+                        curr_state.get("dt"),
+                        reduction[2](curr_state.get("y")),
+                    )
                 )
             )
         else:
-            forces = lambda step, **curr_state: F[step, :] + reduction[1](
-                add_to_RHS(
-                    step,
-                    time_step=curr_state.get("dt"),
-                    disp_resp=reduction[2](curr_state.get("y")),
-                    velc_resp=reduction[2](curr_state.get("ydot")),
-                    accl_resp=reduction[2](curr_state.get("y2dot")),
-                )
-                + magnetic_force(
-                    step,
-                    curr_state.get("dt"),
-                    reduction[2](curr_state.get("y")),
+            forces = lambda step, **curr_state: (
+                F[step, :]
+                + reduction[1](
+                    add_to_RHS(
+                        step,
+                        time_step=curr_state.get("dt"),
+                        disp_resp=reduction[2](curr_state.get("y")),
+                        velc_resp=reduction[2](curr_state.get("ydot")),
+                        accl_resp=reduction[2](curr_state.get("y2dot")),
+                    )
+                    + magnetic_force(
+                        step,
+                        curr_state.get("dt"),
+                        reduction[2](curr_state.get("y")),
+                    )
                 )
             )
 
@@ -2723,8 +2729,8 @@ def _init_ambs_for_integrate(self, dt, **kwargs):
         rotor = deepcopy(self)
 
         if len(magnetic_bearings):
-            magnetic_force = (
-                lambda step, time_step, disp_resp: self.magnetic_bearing_controller(
+            magnetic_force = lambda step, time_step, disp_resp: (
+                self.magnetic_bearing_controller(
                     step, magnetic_bearings, time_step, disp_resp, **kwargs
                 )
             )

ross/tests/test_gear_element.py

@@ -140,7 +140,7 @@ def test_mesh_params(mesh):
     assert mesh.gear_ratio == 1.0
     assert mesh.contact_ratio == 1.7897798668330458
     assert mesh.hertzian_stiffness == 3555868607.9093266
-    assert_allclose(mesh.stiffness, 4.096607e+08, rtol=1e-6, atol=1e-5)
+    assert_allclose(mesh.stiffness, 4.096607e08, rtol=1e-6, atol=1e-5)
     assert_allclose(
         mesh.stiffness_range[:5],
         [