diff options
Diffstat (limited to 'Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py')
| -rwxr-xr-x | Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py b/Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py index fbb108a0e..b97905ee0 100755 --- a/Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py +++ b/Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py @@ -6,7 +6,6 @@ # --- if the correct amount of processors are initialized in AMReX. from mpi4py import MPI -import pywarpx from pywarpx import picmi constants = picmi.constants @@ -128,14 +127,14 @@ new_comm_size = new_comm.size if color == 0: # verify that communicator contains correct number of procs (1) - assert pywarpx.getNProcs() == comm_world_size - 1 - assert pywarpx.getNProcs() == new_comm_size + assert sim.extension.getNProcs() == comm_world_size - 1 + assert sim.extension.getNProcs() == new_comm_size else: # verify that amrex initialized with 1 fewer proc than comm world - assert pywarpx.getNProcs() == comm_world_size - 1 - assert pywarpx.getNProcs() == new_comm_size + assert sim.extension.getNProcs() == comm_world_size - 1 + assert sim.extension.getNProcs() == new_comm_size # verify that amrex proc ranks are offset by -1 from # world comm proc ranks - assert pywarpx.getMyProc() == rank - 1 + assert sim.extension.getMyProc() == rank - 1 |