Apply black formatting to subplots code and tests

Co-authored-by: jnumainville <[email protected]>

Author: Copilot

plugins/plotly-express/src/deephaven/plot/express/plots/subplots.py

@@ -226,13 +226,13 @@ def extract_title_from_figure(fig: Figure | DeephavenFigure) -> str | None:
         if plotly_fig is None:
             return None
         fig = plotly_fig
-    
+
     layout = fig.to_dict().get("layout", {})
     title = layout.get("title")
-    
+
     if title is None:
         return None
-    
+
     # Title can be either a string or a dict with a 'text' key
     if isinstance(title, dict):
         return title.get("text")
@@ -264,21 +264,21 @@ def create_subplot_annotations(
 
     """
     annotations = []
-    
+
     for idx, title in enumerate(titles):
         if not title:  # Skip empty titles
             continue
-            
+
         # Calculate row and col from index (row-major order, but reversed since grid is reversed)
         row = idx // cols
         col = idx % cols
-        
+
         # Calculate x position (center of column)
         x = (col_starts[col] + col_ends[col]) / 2
-        
+
         # Calculate y position (top of row with small offset)
         y = row_ends[row]
-        
+
         annotation = {
             "text": title,
             "showarrow": False,
@@ -288,11 +288,11 @@ def create_subplot_annotations(
             "y": y,
             "xanchor": "center",
             "yanchor": "bottom",
-            "font": {"size": 16}
+            "font": {"size": 16},
         }
-        
+
         annotations.append(annotation)
-    
+
     return annotations
 
 
@@ -380,10 +380,10 @@ def atomic_make_subplots(
 
     # Handle subplot titles
     final_subplot_titles: list[str] = []
-    
+
     if use_existing_titles and subplot_titles is not None:
         raise ValueError("Cannot use both use_existing_titles and subplot_titles")
-    
+
     if use_existing_titles:
         # Extract titles from existing figures
         for fig_row in grid:
@@ -392,16 +392,20 @@ def atomic_make_subplots(
                     final_subplot_titles.append("")
                 else:
                     extracted_title = extract_title_from_figure(fig)
-                    final_subplot_titles.append(extracted_title if extracted_title else "")
+                    final_subplot_titles.append(
+                        extracted_title if extracted_title else ""
+                    )
     elif subplot_titles is not None:
         # Convert to list if tuple
         final_subplot_titles = list(subplot_titles)
-        
+
         # Pad with empty strings if needed
         total_subplots = rows * cols
         if len(final_subplot_titles) < total_subplots:
-            final_subplot_titles.extend([""] * (total_subplots - len(final_subplot_titles)))
-    
+            final_subplot_titles.extend(
+                [""] * (total_subplots - len(final_subplot_titles))
+            )
+
     # Create the custom update function to add annotations and title
     def custom_update_figure(fig: Figure) -> Figure:
         # Add subplot title annotations if any
@@ -415,15 +419,17 @@ def custom_update_figure(fig: Figure) -> Figure:
                 rows,
                 cols,
             )
-            
+
             # Get existing annotations if any
-            existing_annotations = list(fig.layout.annotations) if fig.layout.annotations else []
+            existing_annotations = (
+                list(fig.layout.annotations) if fig.layout.annotations else []
+            )
             fig.update_layout(annotations=existing_annotations + annotations)
-        
+
         # Add overall title if provided
         if title:
             fig.update_layout(title=title)
-        
+
         # Apply user's unsafe_update_figure if provided
         result = unsafe_update_figure(fig)
         return result if result is not None else fig

plugins/plotly-express/test/deephaven/plot/express/plots/test_make_subplots.py

@@ -584,16 +584,16 @@ def test_make_subplots_with_subplot_titles(self):
         layout = charts["plotly"]["layout"]
         self.assertIn("annotations", layout)
         annotations = layout["annotations"]
-        
+
         # Should have 2 annotations for 2 subplot titles
         self.assertEqual(len(annotations), 2)
-        
+
         # Check first annotation
         self.assertEqual(annotations[0]["text"], "Plot 1")
         self.assertEqual(annotations[0]["xref"], "paper")
         self.assertEqual(annotations[0]["yref"], "paper")
         self.assertFalse(annotations[0]["showarrow"])
-        
+
         # Check second annotation
         self.assertEqual(annotations[1]["text"], "Plot 2")
 
@@ -609,7 +609,7 @@ def test_make_subplots_with_empty_subplot_titles(self):
         layout = charts["plotly"]["layout"]
         self.assertIn("annotations", layout)
         annotations = layout["annotations"]
-        
+
         # Should have 2 annotations (empty string skipped)
         self.assertEqual(len(annotations), 2)
         self.assertEqual(annotations[0]["text"], "Plot 1")
@@ -628,7 +628,7 @@ def test_make_subplots_with_use_existing_titles(self):
         layout = charts["plotly"]["layout"]
         self.assertIn("annotations", layout)
         annotations = layout["annotations"]
-        
+
         # Should have 2 annotations
         self.assertEqual(len(annotations), 2)
         self.assertEqual(annotations[0]["text"], "First Chart")
@@ -638,9 +638,9 @@ def test_make_subplots_with_overall_title(self):
         import src.deephaven.plot.express as dx
 
         chart = dx.scatter(self.source, x="X", y="Y")
-        charts = dx.make_subplots(
-            chart, chart, rows=2, title="Overall Title"
-        ).to_dict(self.exporter)
+        charts = dx.make_subplots(chart, chart, rows=2, title="Overall Title").to_dict(
+            self.exporter
+        )
 
         # Check that the overall title was added
         layout = charts["plotly"]["layout"]
@@ -652,18 +652,20 @@ def test_make_subplots_with_subplot_titles_and_overall_title(self):
 
         chart = dx.scatter(self.source, x="X", y="Y")
         charts = dx.make_subplots(
-            chart, chart, rows=2,
+            chart,
+            chart,
+            rows=2,
             subplot_titles=["Plot 1", "Plot 2"],
-            title="Combined Plot"
+            title="Combined Plot",
         ).to_dict(self.exporter)
 
         # Check that both subplot titles and overall title were added
         layout = charts["plotly"]["layout"]
-        
+
         # Check overall title
         self.assertIn("title", layout)
         self.assertEqual(layout["title"], "Combined Plot")
-        
+
         # Check subplot titles
         self.assertIn("annotations", layout)
         annotations = layout["annotations"]
@@ -678,17 +680,17 @@ def test_make_subplots_use_existing_titles_with_grid(self):
         chart2 = dx.scatter(self.source, x="X", y="Y", title="Chart B")
         chart3 = dx.scatter(self.source, x="X", y="Y", title="Chart C")
         chart4 = dx.scatter(self.source, x="X", y="Y", title="Chart D")
-        
+
         grid = [[chart1, chart2], [chart3, chart4]]
-        charts = dx.make_subplots(
-            grid=grid, use_existing_titles=True
-        ).to_dict(self.exporter)
+        charts = dx.make_subplots(grid=grid, use_existing_titles=True).to_dict(
+            self.exporter
+        )
 
         # Check that annotations were added with extracted titles
         layout = charts["plotly"]["layout"]
         self.assertIn("annotations", layout)
         annotations = layout["annotations"]
-        
+
         # Should have 4 annotations
         self.assertEqual(len(annotations), 4)
         self.assertEqual(annotations[0]["text"], "Chart A")
@@ -700,15 +702,17 @@ def test_make_subplots_conflicting_title_options(self):
         import src.deephaven.plot.express as dx
 
         chart = dx.scatter(self.source, x="X", y="Y")
-        
+
         # Should raise error when both use_existing_titles and subplot_titles are provided
         with self.assertRaises(ValueError) as context:
             dx.make_subplots(
-                chart, chart, rows=2,
+                chart,
+                chart,
+                rows=2,
                 subplot_titles=["Plot 1", "Plot 2"],
-                use_existing_titles=True
+                use_existing_titles=True,
             )
-        
+
         self.assertIn("Cannot use both", str(context.exception))