fix numba bug in iqmc_cooper. fix unit test bug due to the passing of…

… particle container
ilhamv · Oct 4, 2024 · 09730ef · 09730ef
1 parent f1dddd8
commit 09730ef
Show file tree

Hide file tree

Showing 3 changed files with 379 additions and 187 deletions.
diff --git a/mcdc/iqmc/iqmc_kernel.py b/mcdc/iqmc/iqmc_kernel.py
@@ -474,7 +474,7 @@ def iqmc_surface_crossing(P_arr, prog):
     if surface["BC"] == BC_VACUUM:
         P["alive"] = False
     elif surface["BC"] == BC_REFLECTIVE:
-        surface_.reflect(P, surface)
+        surface_.reflect(P_arr, surface)
 
     # Need to check new cell later?
     if P["alive"] and not surface["BC"] == BC_REFLECTIVE:

diff --git a/test/unit/src/surface/test_common.py b/test/unit/src/surface/test_common.py
@@ -63,7 +63,7 @@ def test_translate_particle_position():
     t = 6.0
     P = particle(5.0, 3.0, 8.0, t, u, -u, u)
     idx = _get_move_idx(t, surface)
-    _translate_particle_position(P, surface, idx)
+    _translate_particle_position([P], surface, idx)
     x = 5.0 - move_translations[3][0]
     y = 3.0 - move_translations[3][1]
     z = 8.0 - move_translations[3][2]
@@ -79,7 +79,7 @@ def test_translate_particle_position():
     t = 6.5
     P = particle(5.0, 3.0, 8.0, t, u, -u, u)
     idx = _get_move_idx(t, surface)
-    _translate_particle_position(P, surface, idx)
+    _translate_particle_position([P], surface, idx)
     x = 5.0 - move_translations[3][0] - move_velocities[3][0] * (t - time_grid[3])
     y = 3.0 - move_translations[3][1] - move_velocities[3][1] * (t - time_grid[3])
     z = 8.0 - move_translations[3][2] - move_velocities[3][2] * (t - time_grid[3])
@@ -100,7 +100,7 @@ def test_translate_particle_direction():
     t = 6.0
     P = particle(5.0, 3.0, 8.0, t, u, -u, u)
     idx = _get_move_idx(t, surface)
-    _translate_particle_direction(P, speed, surface, idx)
+    _translate_particle_direction([P], speed, surface, idx)
     ux = u - move_velocities[3][0] / speed
     uy = -u - move_velocities[3][1] / speed
     uz = u - move_velocities[3][2] / speed
@@ -116,7 +116,7 @@ def test_translate_particle_direction():
     t = 6.5
     P = particle(5.0, 3.0, 8.0, t, u, -u, u)
     idx = _get_move_idx(t, surface)
-    _translate_particle_direction(P, speed, surface, idx)
+    _translate_particle_direction([P], speed, surface, idx)
     ux = u - move_velocities[3][0] / speed
     uy = -u - move_velocities[3][1] / speed
     uz = u - move_velocities[3][2] / speed