TST: fix slow tests

Author: Gui-FernandesBR

tests/integration/test_environment.py

@@ -160,9 +160,8 @@ def test_nam_atmosphere(mock_show, example_spaceport_env):  # pylint: disable=un
 @pytest.mark.slow
 @patch("matplotlib.pyplot.show")
 def test_rap_atmosphere(mock_show, example_spaceport_env):  # pylint: disable=unused-argument
-    today = date.today()
     now = datetime.now(timezone.utc)
-    example_spaceport_env.set_date((today.year, today.month, today.day, now.hour))
+    example_spaceport_env.set_date((now.year, now.month, now.day, now.hour))
     example_spaceport_env.set_atmospheric_model(type="Forecast", file="RAP")
     assert example_spaceport_env.all_info() is None
 

tests/integration/test_flight.py

@@ -228,7 +228,7 @@ def test_liquid_motor_flight(mock_show, flight_calisto_liquid_modded):  # pylint
 
 @pytest.mark.slow
 @patch("matplotlib.pyplot.show")
-def test_time_overshoot(mock_show, calisto_robust, example_spaceport_env):  # pylint: disable=unused-argument
+def test_time_overshoot_false(mock_show, calisto_robust, example_spaceport_env):  # pylint: disable=unused-argument
     """Test the time_overshoot parameter of the Flight class. This basically
     calls the all_info() method for a simulation without time_overshoot and
     checks if it returns None. It is not testing if the values are correct,
@@ -418,6 +418,19 @@ def test_air_brakes_flight(mock_show, flight_calisto_air_brakes):  # pylint: dis
     assert air_brakes.prints.all() is None
 
 
+@patch("matplotlib.pyplot.show")
+def test_air_brakes_flight_with_overshoot(
+    mock_show, flight_calisto_air_brakes_time_overshoot
+):  # pylint: disable=unused-argument
+    """
+    Same as test_air_brakes_flight but with time_overshoot=True.
+    """
+    test_flight = flight_calisto_air_brakes_time_overshoot
+    air_brakes = test_flight.rocket.air_brakes[0]
+    assert air_brakes.plots.all() is None
+    assert air_brakes.prints.all() is None
+
+
 @patch("matplotlib.pyplot.show")
 def test_initial_solution(mock_show, example_plain_env, calisto_robust):  # pylint: disable=unused-argument
     """Tests the initial_solution option of the Flight class. This test simply

tests/integration/test_plots.py

@@ -74,10 +74,10 @@ def test_compare_flights(mock_show, mock_figure_show, calisto, example_plain_env
     )
 
     calisto.set_rail_buttons(-0.5, 0.2)
-    inclinations = [60, 70, 80, 90]
-    headings = [0, 45, 90, 180]
+    inclinations = [60, 90]
+    headings = [0, 180]
     flights = []
-    # Create (4 * 4) = 16 different flights to be compared
+    # Create (2 * 2) = 4 different flights to be compared
     for heading in headings:
         for inclination in inclinations:
             flight = Flight(