diff --git a/Examples/Physics_applications/capacitive_discharge/PICMI_inputs_1d.py b/Examples/Physics_applications/capacitive_discharge/PICMI_inputs_1d.py index 38b83e46948..01406b7a0fb 100644 --- a/Examples/Physics_applications/capacitive_discharge/PICMI_inputs_1d.py +++ b/Examples/Physics_applications/capacitive_discharge/PICMI_inputs_1d.py @@ -38,7 +38,7 @@ def __init__(self, grid, **kwargs): required_precision=1, **kwargs ) - def initialize_inputs(self): + def solver_initialize_inputs(self): """Grab geometrical quantities from the grid. The boundary potentials are also obtained from the grid using 'warpx_potential_zmin' for the left_voltage and 'warpx_potential_zmax' for the right_voltage. @@ -57,7 +57,7 @@ def initialize_inputs(self): self.grid.potential_zmin = None self.grid.potential_zmax = None - super(PoissonSolver1D, self).initialize_inputs() + super(PoissonSolver1D, self).solver_initialize_inputs() self.nz = self.grid.number_of_cells[0] self.dz = (self.grid.upper_bound[0] - self.grid.lower_bound[0]) / self.nz @@ -421,6 +421,10 @@ def run_sim(self): self.sim.step(self.diag_steps) + if self.pythonsolver: + # confirm that the external solver was run + assert hasattr(self.solver, 'phi') + if libwarpx.amr.ParallelDescriptor.MyProc() == 0: np.save(f'ion_density_case_{self.n+1}.npy', self.ion_density_array) diff --git a/Examples/Physics_applications/capacitive_discharge/PICMI_inputs_2d.py b/Examples/Physics_applications/capacitive_discharge/PICMI_inputs_2d.py index 71e1070ef2f..65baabba605 100755 --- a/Examples/Physics_applications/capacitive_discharge/PICMI_inputs_2d.py +++ b/Examples/Physics_applications/capacitive_discharge/PICMI_inputs_2d.py @@ -75,7 +75,7 @@ def __init__(self, grid, **kwargs): self.phi_wrapper = None self.time_sum = 0.0 - def initialize_inputs(self): + def solver_initialize_inputs(self): """Grab geometrical quantities from the grid. """ self.right_voltage = self.grid.potential_xmax @@ -89,7 +89,7 @@ def initialize_inputs(self): self.grid.potential_zmin = None self.grid.potential_zmax = None - super(PoissonSolverPseudo1D, self).initialize_inputs() + super(PoissonSolverPseudo1D, self).solver_initialize_inputs() self.nx = self.grid.number_of_cells[0] self.nz = self.grid.number_of_cells[1] @@ -359,3 +359,6 @@ def solve(self): ########################## sim.step(max_steps) + +# confirm that the external solver was run +assert hasattr(solver, 'phi')