diff options
Diffstat (limited to 'Examples/Modules/ParticleBoundaryScrape/PICMI_inputs_scrape.py')
| -rwxr-xr-x | Examples/Modules/ParticleBoundaryScrape/PICMI_inputs_scrape.py | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/Examples/Modules/ParticleBoundaryScrape/PICMI_inputs_scrape.py b/Examples/Modules/ParticleBoundaryScrape/PICMI_inputs_scrape.py index 90f526e4b..8009da0cc 100755 --- a/Examples/Modules/ParticleBoundaryScrape/PICMI_inputs_scrape.py +++ b/Examples/Modules/ParticleBoundaryScrape/PICMI_inputs_scrape.py @@ -3,7 +3,6 @@ # --- Input file to test the particle scraper and the Python wrappers # --- to access the buffer of scraped particles. -import pywarpx from pywarpx import picmi ########################## @@ -117,25 +116,25 @@ sim.step(max_steps) from mpi4py import MPI as mpi -my_id = pywarpx.getMyProc() +my_id = sim.extension.getMyProc() -n = pywarpx.get_particle_boundary_buffer_size("electrons", 'eb') +n = sim.extension.get_particle_boundary_buffer_size("electrons", 'eb') print(f"Number of electrons in buffer (proc #{my_id}): {n}") assert n == 612 -scraped_steps = pywarpx.get_particle_boundary_buffer("electrons", 'eb', 'step_scraped', 0) +scraped_steps = sim.extension.get_particle_boundary_buffer("electrons", 'eb', 'step_scraped', 0) for arr in scraped_steps: assert all(arr > 40) -weights = pywarpx.get_particle_boundary_buffer("electrons", 'eb', 'w', 0) +weights = sim.extension.get_particle_boundary_buffer("electrons", 'eb', 'w', 0) n = sum(len(arr) for arr in weights) print(f"Number of electrons in this proc's buffer (proc #{my_id}): {n}") n_sum = mpi.COMM_WORLD.allreduce(n, op=mpi.SUM) assert n_sum == 612 # clear the particle buffer -pywarpx.clearParticleBoundaryBuffer() +sim.extension.clearParticleBoundaryBuffer() # confirm that the buffer was cleared -n = pywarpx.get_particle_boundary_buffer_size("electrons", 'eb') +n = sim.extension.get_particle_boundary_buffer_size("electrons", 'eb') print(f"Number of electrons in buffer (proc #{my_id}): {n}") assert n == 0 |