Apply black formatting

Author: bendudson

xbout/cherab/triangulate.py

@@ -114,7 +114,7 @@ def wall_flux(
 
         Based on the Cherab manual here:
         https://www.cherab.info/demonstrations/radiation_loads/symmetric_power_load.html#symmetric-power-load
-        
+
         Parameters
         ----------
 
@@ -208,7 +208,7 @@ def wall_flux(
             # For checking energy conservation.
             # Revolve this tile around the CYLINDRICAL z-axis to get total power collected by these tiles.
             # Add up all the tile contributions to get total power collected.
-            detector_radius = np.sqrt(detector_center.x ** 2 + detector_center.y ** 2)
+            detector_radius = np.sqrt(detector_center.x**2 + detector_center.y**2)
             observed_total_power = power_density * (
                 y_width * 2 * np.pi * detector_radius
             )

xbout/wall.py

@@ -4,6 +4,7 @@
 
 import numpy as np
 
+
 class AxisymmetricWall:
     def __init__(self, Rs, Zs):
         """
@@ -34,8 +35,9 @@ def __iter__(self):
 
         These pairs define wall segment.
         """
-        return iter(zip(zip(self.Rs, self.Zs),
-                        zip(np.roll(self.Rs, -1), np.roll(self.Zs, -1))))
+        return iter(
+            zip(zip(self.Rs, self.Zs), zip(np.roll(self.Rs, -1), np.roll(self.Zs, -1)))
+        )
 
     def to_polygon(self):
         """
@@ -45,7 +47,7 @@ def to_polygon(self):
         """
         return np.stack((self.Rs, self.Zs), axis=-1)
 
-    def plot(self, linestyle='k-', ax = None):
+    def plot(self, linestyle="k-", ax=None):
         """
         Plot the wall on given axis. If no axis
         is given then a new figure is created.
@@ -64,6 +66,7 @@ def plot(self, linestyle='k-', ax = None):
         ax.plot(self.Rs, self.Zs, linestyle)
         return ax
 
+
 def read_geqdsk(filehandle):
     """
     Read wall geometry from a GEQDSK file.
@@ -86,31 +89,31 @@ def read_geqdsk(filehandle):
     return AxisymmetricWall(data["rlim"], data["zlim"])
 
 
-def read_csv(filehandle, delimiter=','):
+def read_csv(filehandle, delimiter=","):
     """
     Parameters
     ----------
 
     filehandle: File handle
         Must contain two columns, for R and Z coordinates [meters]
-    
+
     delimier : character
         A single character that separates fields
 
     Notes:
-    - Uses the python `csv` module 
+    - Uses the python `csv` module
     """
     import csv
+
     reader = csv.reader(filehandle, delimiter=delimiter)
     Rs = []
     Zs = []
     for row in reader:
         if len(row) == 0:
-            continue # Skip empty rows
+            continue  # Skip empty rows
         if len(row) != 2:
             raise ValueError(f"CSV row should contain two columns: {row}")
         Rs.append(float(row[0]))
         Zs.append(float(row[1]))
 
     return AxisymmetricWall(Rs, Zs)
-