diff options
Diffstat (limited to 'Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py')
-rwxr-xr-x | Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py b/Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py index d71aa3fea..9cfe669ce 100755 --- a/Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py +++ b/Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py @@ -3,7 +3,6 @@ # --- Input file to test the saving of old particle positions import numpy as np -import pywarpx from pywarpx import picmi constants = picmi.constants @@ -110,10 +109,10 @@ sim.step(max_steps - 1) # exist ########################## -assert (pywarpx.get_particle_comp_index('electrons', 'prev_x') > 0) -assert (pywarpx.get_particle_comp_index('electrons', 'prev_z') > 0) +assert (sim.extension.get_particle_comp_index('electrons', 'prev_x') > 0) +assert (sim.extension.get_particle_comp_index('electrons', 'prev_z') > 0) -prev_z_vals = pywarpx.get_particle_arrays( +prev_z_vals = sim.extension.get_particle_arrays( 'electrons', 'prev_z', 0 ) for z_vals in prev_z_vals: |