STY: Reformat with black

Author: C8H10O2

rocketpy/plots/monte_carlo_plots.py

@@ -6,7 +6,12 @@
 import numpy as np
 from matplotlib.transforms import offset_copy
 
-from ..tools import generate_monte_carlo_ellipses, import_optional_dependency, inverted_haversine, haversine
+from ..tools import (
+    generate_monte_carlo_ellipses,
+    import_optional_dependency,
+    inverted_haversine,
+    haversine,
+)
 from .plot_helpers import show_or_save_plot
 
 
@@ -50,7 +55,7 @@ def _get_background_map(self, background, xlim, ylim):
                     "contextily library is required for automatic map background. "
                     "Install it via 'pip install contextily' or 'pip install rocketpy[monte-carlo]'. "
                     "Plotting without background.",
-                    UserWarning
+                    UserWarning,
                 )
                 return None, None
 
@@ -113,11 +118,18 @@ def _get_background_map(self, background, xlim, ylim):
                     dist = (x**2 + y**2) ** 0.5
                     # Calculate bearing: 0 is North (Y), 90 is East (X)
                     bearing = np.degrees(np.arctan2(x, y))
-                    lat, lon = inverted_haversine(origin_lat, origin_lon, dist, bearing, earth_radius)
+                    lat, lon = inverted_haversine(
+                        origin_lat, origin_lon, dist, bearing, earth_radius
+                    )
                     req_lats.append(lat)
                     req_lons.append(lon)
 
-                west, south, east, north = min(req_lons), min(req_lats), max(req_lons), max(req_lats)
+                west, south, east, north = (
+                    min(req_lons),
+                    min(req_lats),
+                    max(req_lons),
+                    max(req_lats),
+                )
 
                 bg, mercator_extent = contextily.bounds2img(
                     west, south, east, north, source=source_provider, ll=True
@@ -127,32 +139,52 @@ def _get_background_map(self, background, xlim, ylim):
                 def mercator_to_wgs84(x, y):
                     r_major = 6378137.0
                     lon = x / r_major * 180.0 / math.pi
-                    lat = (2 * math.atan(math.exp(y / r_major)) - math.pi / 2.0) * 180.0 / math.pi
+                    lat = (
+                        (2 * math.atan(math.exp(y / r_major)) - math.pi / 2.0)
+                        * 180.0
+                        / math.pi
+                    )
                     return lat, lon
 
                 # Convert corners of the fetched image
-                bg_lat_min, bg_lon_min = mercator_to_wgs84(mercator_extent[0], mercator_extent[2]) # Bottom-Left
-                bg_lat_max, bg_lon_max = mercator_to_wgs84(mercator_extent[1], mercator_extent[3]) # Top-Right
+                bg_lat_min, bg_lon_min = mercator_to_wgs84(
+                    mercator_extent[0], mercator_extent[2]
+                )  # Bottom-Left
+                bg_lat_max, bg_lon_max = mercator_to_wgs84(
+                    mercator_extent[1], mercator_extent[3]
+                )  # Top-Right
 
                 # Calculate X/Y meters relative to origin (lat0, lon0) using haversine
                 # X = Distance along longitude (East-West)
                 # Y = Distance along latitude (North-South)
 
                 # Calculate X min (Left)
-                x_min = haversine(origin_lat, origin_lon, origin_lat, bg_lon_min, earth_radius)
-                if bg_lon_min < origin_lon: x_min = -x_min
+                x_min = haversine(
+                    origin_lat, origin_lon, origin_lat, bg_lon_min, earth_radius
+                )
+                if bg_lon_min < origin_lon:
+                    x_min = -x_min
 
                 # Calculate X max (Right)
-                x_max = haversine(origin_lat, origin_lon, origin_lat, bg_lon_max, earth_radius)
-                if bg_lon_max < origin_lon: x_max = -x_max
+                x_max = haversine(
+                    origin_lat, origin_lon, origin_lat, bg_lon_max, earth_radius
+                )
+                if bg_lon_max < origin_lon:
+                    x_max = -x_max
 
                 # Calculate Y min (Bottom)
-                y_min = haversine(origin_lat, origin_lon, bg_lat_min, origin_lon, earth_radius)
-                if bg_lat_min < origin_lat: y_min = -y_min
+                y_min = haversine(
+                    origin_lat, origin_lon, bg_lat_min, origin_lon, earth_radius
+                )
+                if bg_lat_min < origin_lat:
+                    y_min = -y_min
 
                 # Calculate Y max (Top)
-                y_max = haversine(origin_lat, origin_lon, bg_lat_max, origin_lon, earth_radius)
-                if bg_lat_max < origin_lat: y_max = -y_max
+                y_max = haversine(
+                    origin_lat, origin_lon, bg_lat_max, origin_lon, earth_radius
+                )
+                if bg_lat_max < origin_lat:
+                    y_max = -y_max
 
                 return bg, [x_min, x_max, y_min, y_max]
 
@@ -216,7 +248,7 @@ def ellipses(
                     "The image file was not found. Please check the path."
                 ) from e
 
-        bg ,local_extent = None, None
+        bg, local_extent = None, None
         if image is None and background is not None:
             bg, local_extent = self._get_background_map(background, xlim, ylim)
 
@@ -311,19 +343,14 @@ def ellipses(
             )
 
         elif bg is not None and local_extent is not None:
-            plt.imshow(
-                bg,
-                extent=local_extent,
-                zorder=0,
-                interpolation="bilinear"
-            )
+            plt.imshow(bg, extent=local_extent, zorder=0, interpolation="bilinear")
 
         plt.axhline(0, color="black", linewidth=0.5)
         plt.axvline(0, color="black", linewidth=0.5)
         plt.xlim(*xlim)
         plt.ylim(*ylim)
         # Set equal aspect ratio to ensure consistent display regardless of background
-        ax.set_aspect('equal')
+        ax.set_aspect("equal")
 
         if save:
             plt.savefig(
@@ -511,8 +538,7 @@ def ellipses_comparison(
                     "The image file was not found. Please check the path."
                 ) from e
 
-
-        bg ,local_extent = None, None
+        bg, local_extent = None, None
         if image is None and background is not None:
             bg, local_extent = self._get_background_map(background, xlim, ylim)
 
@@ -637,18 +663,13 @@ def ellipses_comparison(
                 ],
             )
         elif bg is not None and local_extent is not None:
-            plt.imshow(
-                bg,
-                extent=local_extent,
-                zorder=0,
-                interpolation="bilinear"
-            )
+            plt.imshow(bg, extent=local_extent, zorder=0, interpolation="bilinear")
 
         plt.axhline(0, color="black", linewidth=0.5)
         plt.axvline(0, color="black", linewidth=0.5)
         plt.xlim(*xlim)
         plt.ylim(*ylim)
-        plt.aspect('equal')
+        plt.aspect("equal")
 
         if save:
             plt.savefig(

tests/integration/simulation/test_monte_carlo_plots_background_kennedy.py

@@ -18,12 +18,12 @@
 
 def create_kennedy_environment():
     """Create an Environment object for Kennedy Space Center.
-    
+
     Kennedy Space Center coordinates:
     - Latitude: 28.5721° N
     - Longitude: -80.6480° W
     - Elevation: ~3 meters
-    
+
     Returns
     -------
     Environment
@@ -44,7 +44,7 @@ def create_kennedy_environment():
 
 def create_simulated_monte_carlo_data():
     """Create a MonteCarlo object with simulated results data.
-    
+
     Returns
     -------
     MonteCarlo
@@ -62,7 +62,6 @@ def create_simulated_monte_carlo_data():
         longitude=None,
     )
 
-
     from unittest.mock import MagicMock
 
     stochastic_rocket = MagicMock()
@@ -100,7 +99,7 @@ def create_simulated_monte_carlo_data():
 
 def test_all_background_options():
     """Test all background map options and save images.
-    
+
     This function tests:
     - None (no background)
     - satellite
@@ -157,4 +156,4 @@ def test_all_background_options():
 
 
 if __name__ == "__main__":
-    test_all_background_options()
+    test_all_background_options()

tests/unit/simulation/test_monte_carlo_plots_background.py

@@ -138,8 +138,9 @@ def test_ellipses_image_takes_precedence_over_background(
         # Test that when both image and background are provided, image takes precedence
         # This should not attempt to download background map
         import numpy as np
+
         mock_image = np.zeros((100, 100, 3), dtype=np.uint8)  # RGB image
-        
+
         with patch("imageio.imread") as mock_imread:
             mock_imread.return_value = mock_image
             result = monte_carlo_calisto_pre_loaded.plots.ellipses(
@@ -171,7 +172,9 @@ def __init__(self):
     mock_monte_carlo = MockMonteCarlo()
     plots = _MonteCarloPlots(mock_monte_carlo)
 
-    with pytest.raises(ValueError, match="MonteCarlo object must have an 'environment' attribute"):
+    with pytest.raises(
+        ValueError, match="MonteCarlo object must have an 'environment' attribute"
+    ):
         plots.ellipses(background="satellite")
 
 
@@ -191,12 +194,16 @@ def test_ellipses_background_no_latitude_longitude():
 
     plots = _MonteCarloPlots(mock_monte_carlo)
 
-    with pytest.raises(ValueError, match="Environment must have 'latitude' and 'longitude' attributes"):
+    with pytest.raises(
+        ValueError, match="Environment must have 'latitude' and 'longitude' attributes"
+    ):
         plots.ellipses(background="satellite")
 
 
 @patch("matplotlib.pyplot.show")
-def test_ellipses_background_contextily_not_installed(mock_show, monte_carlo_calisto_pre_loaded):
+def test_ellipses_background_contextily_not_installed(
+    mock_show, monte_carlo_calisto_pre_loaded
+):
     """Test that a warning is issued when contextily is not installed.
 
     Parameters
@@ -215,19 +222,28 @@ def mock_import_optional_dependency(name):
                 raise ImportError("No module named 'contextily'")
             return original_import(name)
 
-        with patch("rocketpy.plots.monte_carlo_plots.import_optional_dependency", side_effect=mock_import_optional_dependency):
+        with patch(
+            "rocketpy.plots.monte_carlo_plots.import_optional_dependency",
+            side_effect=mock_import_optional_dependency,
+        ):
             with warnings.catch_warnings(record=True) as w:
                 warnings.simplefilter("always")
-                result = monte_carlo_calisto_pre_loaded.plots.ellipses(background="satellite")
+                result = monte_carlo_calisto_pre_loaded.plots.ellipses(
+                    background="satellite"
+                )
                 assert result is None
                 assert len(w) > 0
-                assert any("contextily" in str(warning.message).lower() for warning in w)
+                assert any(
+                    "contextily" in str(warning.message).lower() for warning in w
+                )
     finally:
         _post_test_file_cleanup()
 
 
 @patch("matplotlib.pyplot.show")
-def test_ellipses_background_with_custom_xlim_ylim(mock_show, monte_carlo_calisto_pre_loaded):
+def test_ellipses_background_with_custom_xlim_ylim(
+    mock_show, monte_carlo_calisto_pre_loaded
+):
     """Test using background with custom xlim and ylim.
 
     Parameters
@@ -270,4 +286,3 @@ def test_ellipses_background_save(mock_show, monte_carlo_calisto_pre_loaded):
         assert os.path.exists("monte_carlo_test.png")
     finally:
         _post_test_file_cleanup()
-