Fix tests

Author: waltsims

tests/test_kspaceFirstOrder3D_state.py

@@ -39,18 +39,18 @@ def make_simulation_parameters(directory: Path):
     # define a binary planar sensor
     sensor = kSensor()
     sensor_mask_array = np.zeros(N)
-    sensor_mask_array[0, :, :] = 1 # Corrected to be a plane for 3D
+    sensor_mask_array[0, :, :] = 1  # Corrected to be a plane for 3D
     sensor.mask = sensor_mask_array
 
     input_filename = directory / "kwave_input.h5"
     output_filename = directory / "kwave_output.h5"
     checkpoint_filename = directory / "kwave_checkpoint.h5"
 
     simulation_options = SimulationOptions(
-        save_to_disk=True, # Must be true for kspaceFirstOrder3D
+        save_to_disk=True,  # Must be true for kspaceFirstOrder3D
         pml_size=PML_size,
         pml_inside=False,
-        smooth_p0=False, # p0 is already smoothed
+        smooth_p0=False,  # p0 is already smoothed
         data_cast="single",
         input_filename=input_filename,
         output_filename=output_filename,
@@ -59,10 +59,10 @@ def make_simulation_parameters(directory: Path):
     checkpoint_timesteps = 300
 
     execution_options = SimulationExecutionOptions(
-        is_gpu_simulation=False, # Assuming CPU for basic test
+        is_gpu_simulation=False,  # Assuming CPU for basic test
         checkpoint_file=checkpoint_filename,
         checkpoint_timesteps=checkpoint_timesteps,
-        verbose_level=0 # Keep test output clean
+        verbose_level=0,  # Keep test output clean
     )
     return kgrid, medium, source, sensor, simulation_options, execution_options
 
@@ -75,13 +75,20 @@ def test_kspaceFirstOrder3D_input_state_preservation():
         # Store original states of critical attributes for comparison
         original_source_p0 = deepcopy(source.p0)
         original_sensor_mask = deepcopy(sensor.mask)
-        
+
         # If source.p or source.u were time-varying, store their initial states too.
         # For this test, p0 is the main source attribute.
 
         # First run
         try:
-            _ = kspaceFirstOrder3D(kgrid, medium, source, sensor, simulation_options, execution_options)
+            _ = kspaceFirstOrder3D(
+                kgrid=kgrid,
+                medium=medium,
+                source=source,
+                sensor=sensor,
+                simulation_options=simulation_options,
+                execution_options=execution_options,
+            )
         except Exception as e:
             pytest.fail(f"First call to kspaceFirstOrder3D failed: {e}")
 
@@ -95,13 +102,20 @@ def test_kspaceFirstOrder3D_input_state_preservation():
         simulation_options_run2 = deepcopy(simulation_options)
         simulation_options_run2.input_filename = tmpdir / "kwave_input_run2.h5"
         simulation_options_run2.output_filename = tmpdir / "kwave_output_run2.h5"
-        
+
         execution_options_run2 = deepcopy(execution_options)
-        if execution_options_run2.checkpoint_file: # Only change if it exists
+        if execution_options_run2.checkpoint_file:  # Only change if it exists
             execution_options_run2.checkpoint_file = tmpdir / "kwave_checkpoint_run2.h5"
 
         try:
-            _ = kspaceFirstOrder3D(kgrid, medium, source, sensor, simulation_options_run2, execution_options_run2)
+            _ = kspaceFirstOrder3D(
+                kgrid=kgrid,
+                medium=medium,
+                source=source,
+                sensor=sensor,
+                simulation_options=simulation_options_run2,
+                execution_options=execution_options_run2,
+            )
         except Exception as e:
             pytest.fail(f"Second call to kspaceFirstOrder3D with original objects failed: {e}")