diff options
Diffstat (limited to 'Examples/Tests/particle_boundary_scrape/PICMI_inputs_scrape.py')
| -rwxr-xr-x | Examples/Tests/particle_boundary_scrape/PICMI_inputs_scrape.py | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/Examples/Tests/particle_boundary_scrape/PICMI_inputs_scrape.py b/Examples/Tests/particle_boundary_scrape/PICMI_inputs_scrape.py index 8009da0cc..66a1c0c39 100755 --- a/Examples/Tests/particle_boundary_scrape/PICMI_inputs_scrape.py +++ b/Examples/Tests/particle_boundary_scrape/PICMI_inputs_scrape.py @@ -3,7 +3,9 @@ # --- Input file to test the particle scraper and the Python wrappers # --- to access the buffer of scraped particles. -from pywarpx import picmi +import numpy as np + +from pywarpx import particle_containers, picmi ########################## # numerics parameters @@ -118,23 +120,24 @@ from mpi4py import MPI as mpi my_id = sim.extension.getMyProc() -n = sim.extension.get_particle_boundary_buffer_size("electrons", 'eb') +particle_buffer = particle_containers.ParticleBoundaryBufferWrapper() + +n = particle_buffer.get_particle_boundary_buffer_size("electrons", 'eb') print(f"Number of electrons in buffer (proc #{my_id}): {n}") assert n == 612 -scraped_steps = sim.extension.get_particle_boundary_buffer("electrons", 'eb', 'step_scraped', 0) +scraped_steps = particle_buffer.get_particle_boundary_buffer("electrons", 'eb', 'step_scraped', 0) for arr in scraped_steps: - assert all(arr > 40) + assert all(np.array(arr, copy=False) > 40) -weights = sim.extension.get_particle_boundary_buffer("electrons", 'eb', 'w', 0) +weights = particle_buffer.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 -sim.extension.clearParticleBoundaryBuffer() +particle_buffer.clear_buffer() # confirm that the buffer was cleared -n = sim.extension.get_particle_boundary_buffer_size("electrons", 'eb') +n = particle_buffer.get_particle_boundary_buffer_size("electrons", 'eb') print(f"Number of electrons in buffer (proc #{my_id}): {n}") assert n == 0 |