diff options
Diffstat (limited to 'Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py')
| -rwxr-xr-x | Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py | 22 |
1 files changed, 10 insertions, 12 deletions
diff --git a/Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py b/Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py index 6ed144dbc..2162236eb 100755 --- a/Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py +++ b/Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py @@ -1,10 +1,9 @@ #!/usr/bin/env python3 - -from pywarpx import picmi -import numpy as np - import argparse import sys +import numpy as np +import pywarpx +from pywarpx import picmi, callbacks # Create the parser and add the argument parser = argparse.ArgumentParser() @@ -106,11 +105,10 @@ sim.initialize_warpx() # python particle data access ########################## -from pywarpx import _libwarpx, callbacks -_libwarpx.add_real_comp('electrons', 'newPid') +pywarpx.add_real_comp('electrons', 'newPid') -my_id = _libwarpx.libwarpx.warpx_getMyProc() +my_id = pywarpx.getMyProc() def add_particles(): @@ -124,7 +122,7 @@ def add_particles(): w = np.ones(nps) * 2.0 newPid = 5.0 - _libwarpx.add_particles( + pywarpx.add_particles( species_name='electrons', x=x, y=y, z=z, ux=ux, uy=uy, uz=uz, w=w, newPid=newPid, unique_particles=args.unique ) @@ -142,11 +140,11 @@ sim.step(max_steps - 1) # are properly set ########################## -assert (_libwarpx.get_particle_count('electrons') == 270 / (2 - args.unique)) -assert (_libwarpx.get_particle_comp_index('electrons', 'w') == 0) -assert (_libwarpx.get_particle_comp_index('electrons', 'newPid') == 4) +assert (pywarpx.get_particle_count('electrons') == 270 / (2 - args.unique)) +assert (pywarpx.get_particle_comp_index('electrons', 'w') == 0) +assert (pywarpx.get_particle_comp_index('electrons', 'newPid') == 4) -new_pid_vals = _libwarpx.get_particle_arrays( +new_pid_vals = pywarpx.get_particle_arrays( 'electrons', 'newPid', 0 ) for vals in new_pid_vals: |