MNT: Refactor tank testing and improve file structure.

Author: phmbressan

rocketpy/motors/tank_geometry.py

@@ -1,6 +1,7 @@
 from functools import cached_property
 
 import numpy as np
+
 from ..mathutils.function import Function, funcify_method
 from ..mathutils.piecewise_function import PiecewiseFunction
 from ..plots.tank_geometry_plots import _TankGeometryPlots

tests/unit/test_tank.py

@@ -348,18 +348,4 @@ def expected_gas_inertia(t):
         tank.liquid_inertia.y_array,
         atol=1e-3,
         rtol=1e-2,
-    )
-    assert np.allclose(
-        expected_gas_inertia(times), tank.gas_inertia.y_array, atol=1e-3, rtol=1e-2
-    )
-    assert np.allclose(
-        expected_liquid_inertia(times) + expected_gas_inertia(times),
-        tank.inertia.y_array,
-        atol=1e-3,
-        rtol=1e-2,
-    )
-
-
-@patch("matplotlib.pyplot.show")
-def test_tank_geometry_plots_info(mock_show):  # pylint: disable=unused-argument
-    assert TankGeometry({(0, 5): 1}).plots.all() is None
+    )