black formatting

elastica/__init__.py

@@ -5,7 +5,7 @@
 try:
     import numba
 
-    if 'IMPORT_TEST_NUMPY' in os.environ:
+    if "IMPORT_TEST_NUMPY" in os.environ:
         # This is used to test if numpy version of classes can be imported.
         raise ImportError
 

elastica/_elastica_numba/_timestepper/_explicit_steppers.py

@@ -73,5 +73,3 @@ def _first_stage(self, System, Memory, time, dt):
     def _first_update(self, System, Memory, time, dt):
         System.state += dt * System(time, dt)
         return time + dt
-
-

elastica/timestepper/__init__.py

@@ -25,5 +25,3 @@
         extend_stepper_interface,
         integrate,
     )
-
-

elastica/timestepper/explicit_steppers.py

@@ -150,7 +150,6 @@ def n_stages(self):
     )
 
 
-
 class StatefulRungeKutta4(_StatefulStepper):
     """
     Stores all states of Rk within the time-stepper. Works as long as the states

elastica/timestepper/symplectic_steppers.py

@@ -200,5 +200,3 @@ def n_stages(self):
         PositionVerlet,
         PEFRL,
     )
-
-

examples/Visualization/ContinuumSnakeVisualisation/continuum_snake.py

@@ -20,9 +20,7 @@ class SnakeSimulator(BaseSystemCollection, Constraints, Forcing, CallBacks):
     pass
 
 
-def run_snake(
-    b_coeff, SAVE_RESULTS=False
-):
+def run_snake(b_coeff, SAVE_RESULTS=False):
 
     snake_sim = SnakeSimulator()
 
@@ -153,8 +151,6 @@ def make_callback(self, system, time, current_step: int):
     t_coeff_optimized = np.array([17.4, 48.5, 5.4, 14.7, 0.97])
 
     # run the simulation
-    pp_list = run_snake(
-        t_coeff_optimized, SAVE_RESULTS
-    )
+    pp_list = run_snake(t_coeff_optimized, SAVE_RESULTS)
 
     print("Datafile Created")

examples/Visualization/ContinuumSnakeVisualisation/continuum_snake_render.py

@@ -42,20 +42,20 @@
     location=[15.0, 10.5, -15.0],
     angle=30,
     look_at=[4.0, 2.7, 2.0],
-    name='diag'
+    name="diag",
 )
 stages.add_camera(
     # Add top viewpoint
     location=[0, 15, 3],
     angle=30,
     look_at=[0.0, 0, 3],
     sky=[-1, 0, 0],
-    name='top'
+    name="top",
 )
 stages.add_light(
     # Sun light
     position=[1500, 2500, -1000],
-    color='White',
+    color="White",
     camera_id=-1,
 )
 stages.add_light(
@@ -75,7 +75,7 @@
 # Externally Including Files (USER DEFINE)
 # If user wants to include other POVray objects such as grid or coordinate axes,
 # objects can be defined externally and included separately.
-included = ['../default.inc']
+included = ["../default.inc"]
 
 # Multiprocessing Configuration (USER DEFINE)
 MULTIPROCESSING = True
@@ -122,10 +122,10 @@
     # Rendering
     # For each frame, a 'pov' script file is generated in OUTPUT_IMAGE_DIR directory.
     batch = []
-    for view_name in stage_scripts.keys(): # Make Directory
+    for view_name in stage_scripts.keys():  # Make Directory
         output_path = os.path.join(OUTPUT_IMAGES_DIR, view_name)
         os.makedirs(output_path, exist_ok=True)
-    for frame_number in tqdm(range(total_frame), desc='Scripting'):
+    for frame_number in tqdm(range(total_frame), desc="Scripting"):
         for view_name, stage_script in stage_scripts.items():
             output_path = os.path.join(OUTPUT_IMAGES_DIR, view_name)
 
@@ -139,31 +139,31 @@
             rod_object = pyelastica_rod(
                 x=xs[frame_number],
                 r=base_radius[frame_number],
-                color='rgb<0.45,0.39,1>'
+                color="rgb<0.45,0.39,1>",
             )
             script.append(rod_object)
             pov_script = "\n".join(script)
 
             # Write .pov script file
             file_path = os.path.join(output_path, "frame_{:04d}".format(frame_number))
-            with open(file_path+'.pov', "w+") as f:
+            with open(file_path + ".pov", "w+") as f:
                 f.write(pov_script)
             batch.append(file_path)
 
     # Process POVray
     # For each frames, a 'png' image file is generated in OUTPUT_IMAGE_DIR directory.
-    pbar = tqdm(total=len(batch), desc='Rendering')  # Progress Bar
+    pbar = tqdm(total=len(batch), desc="Rendering")  # Progress Bar
     if MULTIPROCESSING:
         func = partial(
-                render,
-                width=WIDTH,
-                height=HEIGHT,
-                display=DISPLAY_FRAMES,
-                pov_thread=THREAD_PER_AGENT
-            )
+            render,
+            width=WIDTH,
+            height=HEIGHT,
+            display=DISPLAY_FRAMES,
+            pov_thread=THREAD_PER_AGENT,
+        )
         with Pool(NUM_AGENT) as p:
             for message in p.imap_unordered(func, batch):
-                #(TODO) POVray error within child process could be an issue
+                # (TODO) POVray error within child process could be an issue
                 pbar.update()
     else:
         for filename in batch:
@@ -172,16 +172,19 @@
                 width=WIDTH,
                 height=HEIGHT,
                 display=DISPLAY_FRAMES,
-                pov_thread=multiprocessing.cpu_count()
+                pov_thread=multiprocessing.cpu_count(),
             )
             pbar.update()
 
     # Create Video using moviepy
     for view_name in stage_scripts.keys():
         imageset_path = os.path.join(OUTPUT_IMAGES_DIR, view_name)
-        imageset = [os.path.join(imageset_path, path)
-                for path in os.listdir(imageset_path) if path[-3:]=='png']
+        imageset = [
+            os.path.join(imageset_path, path)
+            for path in os.listdir(imageset_path)
+            if path[-3:] == "png"
+        ]
         imageset.sort()
-        filename = OUTPUT_FILENAME + '_' + view_name + '.mp4'
+        filename = OUTPUT_FILENAME + "_" + view_name + ".mp4"
         clip = ImageSequenceClip(imageset, fps=FPS)
         clip.write_videofile(filename, fps=FPS)

examples/Visualization/_povmacros.py

@@ -12,7 +12,7 @@
 def pyelastica_rod(
     x,
     r,
-    color='rgb<0.45,0.39,1>',
+    color="rgb<0.45,0.39,1>",
     transmit=0.0,
     interpolation="linear_spline",
     deform=None,
@@ -73,13 +73,7 @@ def pyelastica_rod(
 
 
 def render(
-    filename,
-    width,
-    height,
-    antialias="on",
-    quality=11,
-    display="Off",
-    pov_thread=4
+    filename, width, height, antialias="on", quality=11, display="Off", pov_thread=4
 ):
     """Rendering frame
 
@@ -115,8 +109,8 @@ def render(
     """
 
     # Define script path and image path
-    script_file = filename + '.pov'
-    image_file = filename + '.png'
+    script_file = filename + ".pov"
+    image_file = filename + ".png"
 
     # Run Povray as subprocess
     cmds = [
@@ -128,7 +122,7 @@ def render(
         f"Work_Threads={pov_thread}",
         f"Antialias={antialias}",
         f"Quality={quality}",
-        f"Display={display}"
+        f"Display={display}",
     ]
     process = subprocess.Popen(
         cmds, stderr=subprocess.PIPE, stdin=subprocess.PIPE, stdout=subprocess.PIPE
@@ -266,7 +260,7 @@ class StageObject:
         """
 
         def __init__(self):
-            self.str = ''
+            self.str = ""
             self.update_script()
 
         def update_script(self):