diff options
Diffstat (limited to 'Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py')
| -rwxr-xr-x | Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py b/Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py index 6df8d676c..ff3c0704b 100755 --- a/Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py +++ b/Examples/Tests/ParticleDataPython/PICMI_inputs_2d.py @@ -3,7 +3,6 @@ import argparse import sys import numpy as np -import pywarpx from pywarpx import callbacks, picmi # Create the parser and add the argument @@ -107,9 +106,9 @@ sim.initialize_warpx() ########################## -pywarpx.add_real_comp('electrons', 'newPid') +sim.extension.add_real_comp('electrons', 'newPid') -my_id = pywarpx.getMyProc() +my_id = sim.extension.getMyProc() def add_particles(): @@ -123,7 +122,7 @@ def add_particles(): w = np.ones(nps) * 2.0 newPid = 5.0 - pywarpx.add_particles( + sim.extension.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 ) @@ -141,11 +140,11 @@ sim.step(max_steps - 1) # are properly set ########################## -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) +assert (sim.extension.get_particle_count('electrons') == 270 / (2 - args.unique)) +assert (sim.extension.get_particle_comp_index('electrons', 'w') == 0) +assert (sim.extension.get_particle_comp_index('electrons', 'newPid') == 4) -new_pid_vals = pywarpx.get_particle_arrays( +new_pid_vals = sim.extension.get_particle_arrays( 'electrons', 'newPid', 0 ) for vals in new_pid_vals: |