Merge pull request #1138 from ViniciusTxc3/dev_pint_plot_deflected_shape

get speed in pint library for plot_deflected_shape function
petrobras · Jan 16, 2025 · 8df3e1c · 8df3e1c
2 parents 695e6fe + aa1a52a
commit 8df3e1c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/ross/results.py b/ross/results.py
@@ -4051,6 +4051,7 @@ def plot_bending_moment(
 
         return fig
 
+    @check_units
     def plot_deflected_shape(
         self,
         speed,
@@ -4074,7 +4075,7 @@ def plot_deflected_shape(
 
         Parameters
         ----------
-        speed : float
+        speed : float, pint.Quantity
             The rotor rotation speed. Must be an element from the speed_range argument
             passed to the class (rad/s).
         samples : int, optional

diff --git a/ross/rotor_assembly.py b/ross/rotor_assembly.py
@@ -1247,15 +1247,15 @@ def A(self, speed=0, frequency=None, synchronous=False):
         Examples
         --------
         >>> rotor = rotor_example()
-        >>> np.round(rotor.A()[75:83, :2])
+        >>> np.round(rotor.A()[75:83, :2]) + 0.
         array([[     0.,  10927.],
-               [-10924.,     -0.],
-               [    -0.,     -0.],
+               [-10924.,      0.],
+               [     0.,      0.],
                [  -174.,      0.],
-               [    -0.,   -174.],
-               [    -0.,     -0.],
-               [    -0.,  10723.],
-               [-10719.,     -0.]])
+               [     0.,   -174.],
+               [     0.,      0.],
+               [     0.,  10723.],
+               [-10719.,      0.]])
         """
         if frequency is None:
             frequency = speed