[skip ci] Apply formatting changes

examples/example_plot_plasma_source_position.py

@@ -1,4 +1,3 @@
-
 import openmc_source_plotter as osp
 from openmc_plasma_source import TokamakSource
 

examples/example_plot_source_direction.py

@@ -1,4 +1,3 @@
-
 import openmc_source_plotter as osp
 import openmc
 

examples/example_plot_source_energy.py

@@ -1,4 +1,3 @@
-
 import openmc_source_plotter as osp
 import openmc
 import numpy as np

examples/example_plot_source_position.py

@@ -1,4 +1,3 @@
-
 import openmc_source_plotter as osp
 import openmc
 
@@ -21,8 +20,7 @@
 
 # plots the particle energy distribution
 plot = osp.plot_source_position(
-    source = my_source,
-    openmc_exec="/home/jshim/miniconda3/envs/openmc_0_11_0/bin/openmc"
+    source=my_source, openmc_exec="/home/jshim/miniconda3/envs/openmc_0_11_0/bin/openmc"
 )
 
 plot.show()

examples/example_plot_two_source_energies.py

@@ -1,4 +1,3 @@
-
 import openmc_source_plotter as osp
 import openmc
 import numpy as np

openmc_source_plotter/__init__.py

@@ -3,4 +3,3 @@
 from .core import plot_source_direction
 from .core import plot_source_energy
 from .core import plot_source_position
-

openmc_source_plotter/core.py

@@ -19,7 +19,7 @@ def plot_source_energy(
     energy_bins: np.array = np.linspace(0, 20e6, 50),
 ):
     """makes a plot of the energy distribution OpenMC source(s)
-    
+
     Args:
         source: The openmc.Source object or list of openmc.Source objects to plot.
         number_of_particles: The number of source samples to obtain, more will
@@ -36,10 +36,10 @@ def plot_source_energy(
     for single_source in source:
         tmp_filename = tempfile.mkstemp(suffix=".h5", prefix=f"openmc_source_")[1]
         create_initial_particles(
-            source= single_source,
+            source=single_source,
             number_of_particles=number_of_particles,
             openmc_exec=openmc_exec,
-            output_source_filename=tmp_filename
+            output_source_filename=tmp_filename,
         )
 
         print("getting particle data", tmp_filename)
@@ -77,7 +77,7 @@ def plot_source_position(
     openmc_exec="openmc",
 ):
     """makes a plot of the initial creation postions of an OpenMC source(s)
-    
+
     Args:
         source: The openmc.Source object or list of openmc.Source objects to plot.
         number_of_particles: The number of source samples to obtain.
@@ -92,10 +92,10 @@ def plot_source_position(
     for single_source in source:
         tmp_filename = tempfile.mkstemp(suffix=".h5", prefix=f"openmc_source_")[1]
         create_initial_particles(
-            source= single_source,
+            source=single_source,
             number_of_particles=number_of_particles,
             openmc_exec=openmc_exec,
-            output_source_filename=tmp_filename
+            output_source_filename=tmp_filename,
         )
 
         data = get_particle_data(tmp_filename)
@@ -128,7 +128,7 @@ def plot_source_direction(
     openmc_exec="openmc",
 ):
     """makes a plot of the initial creation directions of the particle source
-    
+
     Args:
         source: The openmc.Source object or list of openmc.Source objects to plot.
         number_of_particles: The number of source samples to obtain.
@@ -142,10 +142,10 @@ def plot_source_direction(
     for single_source in source:
         tmp_filename = tempfile.mkstemp(suffix=".h5", prefix=f"openmc_source_")[1]
         create_initial_particles(
-            source= single_source,
+            source=single_source,
             number_of_particles=number_of_particles,
             openmc_exec=openmc_exec,
-            output_source_filename=tmp_filename
+            output_source_filename=tmp_filename,
         )
         data = get_particle_data(tmp_filename)
 

tests/test_core.py

@@ -6,6 +6,7 @@
 import numpy as np
 import plotly.graph_objects as go
 
+
 class TestUtils(unittest.TestCase):
     def setUp(self):
 
@@ -50,15 +51,15 @@ def test_energy_plot(self):
     def test_position_plot(self):
 
         plot = osp.plot_source_position(
-            source = self.my_source,
-            openmc_exec=self.openmc_exec_dict[self.current_computer]
+            source=self.my_source,
+            openmc_exec=self.openmc_exec_dict[self.current_computer],
         )
         assert isinstance(plot, go.Figure)
 
     def test_direction_plot(self):
         plot = osp.plot_source_direction(
             source=self.my_source,
             number_of_particles=100,
-            openmc_exec=self.openmc_exec_dict['laptop'],
+            openmc_exec=self.openmc_exec_dict["laptop"],
         )
         assert isinstance(plot, go.Figure)