Run "black" python formatter

tests/test_animate.py

@@ -36,7 +36,6 @@ class TestAnimate:
     """
 
     def test_animate2D(self, create_single_test_dataset):
-
         ds = create_single_test_dataset
         save_dir = "."
         animation = ds["electrons"].feltor.animate2D(save_as="%s/testxy" % save_dir)
@@ -52,7 +51,6 @@ def test_animate2D(self, create_single_test_dataset):
         os.system("rm testxy.gif")
 
     def test_animate1D(self, create_single_test_dataset):
-
         ds = create_single_test_dataset
 
         save_dir = "."
@@ -68,7 +66,6 @@ def test_animate1D(self, create_single_test_dataset):
         os.system("rm test.gif")
 
     def test_animate_list(self, create_single_test_dataset):
-
         ds = create_single_test_dataset
 
         animation = ds.feltor.animate_list([ds["electrons"], ds["electrons"].isel(y=1)])

xfeltor/__init__.py

@@ -19,4 +19,5 @@
 from .load import open_feltordataset
 from .feltordataarray import FeltorDataArrayAccessor
 from .feltordataset import FeltorDatasetAccessor
-#Why is plotting not loaded? Should it be hidden from help(xfeltor)?
+
+# Why is plotting not loaded? Should it be hidden from help(xfeltor)?

xfeltor/feltordataarray.py

@@ -10,6 +10,7 @@
 # The README says the plotting routines are from xBOUT. How far is that copy verbose and do we have to mention them here (License)?
 # Or are the methods just named like in the xBOUT library?
 
+
 # A mechanism to extend the xarray.DataArray class by registering a custom property
 # https://docs.xarray.dev/en/stable/generated/xarray.register_dataarray_accessor.html
 @register_dataarray_accessor("feltor")

xfeltor/feltordataset.py

@@ -8,8 +8,9 @@
 import numpy as np
 from .plotting import _add_controls
 
+
 # A mechanism to extend the xarray.Dataset class by registering a custom property
-#https://docs.xarray.dev/en/stable/internals/extending-xarray.html
+# https://docs.xarray.dev/en/stable/internals/extending-xarray.html
 @xr.register_dataset_accessor("feltor")
 class FeltorDatasetAccessor:
     """Contains FELTOR-specific methods to use on FELTOR datasets opened using
@@ -28,6 +29,7 @@ class FeltorDatasetAccessor:
     )                                       # use its methods
 
     """
+
     def __init__(self, ds):
         self.data = ds
 
@@ -44,7 +46,7 @@ def __str__(self):
         print(ds.feltor)
         # same as above but will not print the "inputfile" attribute
         """
-        ds = self.data.copy() # a shallow copy ...
+        ds = self.data.copy()  # a shallow copy ...
         del ds.attrs["inputfile"]
         styled = partial(prettyformat, indent=4, compact=False)
         return "<xfeltor.FeltorDataset>" + "\n{}\n".format(styled(ds))

xfeltor/plotting.py

@@ -46,7 +46,7 @@ def _normalise_time_coord(time_values):
     tmax = time_values.max()
     if tmax < 1.0e-2 or tmax > 1.0e6:
         scale_pow = int(np.floor(np.log10(tmax)))
-        scale_factor = 10 ** scale_pow
+        scale_factor = 10**scale_pow
         time_values = time_values / scale_factor
         suffix = f"e{scale_pow}"
     else:
@@ -249,8 +249,12 @@ def animate_pcolormesh(
             UserWarning,
         )
         pcolormesh_block = amp.blocks.Pcolormesh(
-            x_values, y_values, image_data, ax=ax, **kwargs
-# shading parameter triggers error when trying to set manually
+            x_values,
+            y_values,
+            image_data,
+            ax=ax,
+            **kwargs
+            # shading parameter triggers error when trying to set manually
         )
 
     if animate: