Debugging

ax3l · Feb 16, 2024 · 01f6b58 · 01f6b58
1 parent f0b4a64
commit 01f6b58
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/amrex/StructOfArrays.py b/src/amrex/StructOfArrays.py
@@ -114,6 +114,7 @@ def soa_to_numpy(self, copy=False):
         real_comp_names = self.soa_real_comps(self.num_real_comps, rotate=False)
 
     for idx_real in range(self.num_real_comps):
+        print(idx_real, real_comp_names[idx_real], self.get_real_data(idx_real).size())
         soa_view.real[real_comp_names[idx_real]] = self.get_real_data(
             idx_real
         ).to_numpy(copy=copy)

diff --git a/tests/test_particleContainer.py b/tests/test_particleContainer.py
@@ -353,6 +353,7 @@ def test_per_cell(empty_particle_container, std_geometry, std_particle):
 def test_soa_pc_numpy(soa_particle_container, Npart):
     """Used in docs/source/usage/compute.rst"""
     pc = soa_particle_container
+    assert pc.number_of_particles_at_level(0) == Npart
 
     class Config:
         have_gpu = False
@@ -397,6 +398,7 @@ class Config:
 def test_pc_numpy(particle_container, Npart):
     """Used in docs/source/usage/compute.rst"""
     pc = particle_container
+    assert pc.number_of_particles_at_level(0) == Npart
 
     class Config:
         have_gpu = False