Correction of a bug in test019 when trying to use a number of MU equa…

…l to 0
OpenGATE · Jul 5, 2024 · 75d510e · 75d510e
1 parent ec6dcc3
commit 75d510e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/opengate/tests/src/test019_linac_elekta_versa_with_mlc_rt_plan.py b/opengate/tests/src/test019_linac_elekta_versa_with_mlc_rt_plan.py
@@ -186,12 +186,15 @@ def validation_test_19_rt_plan(
     nb_part = 750000
     z_linac = linac.size[2]
     rt_plan_parameters = rtplan.read(str(paths.data / "DICOM_RT_plan.dcm"))
-    l_cp = [np.random.randint(0, len(rt_plan_parameters["jaws 1"]), 1)[0]]
+    MU = 0
+    while MU == 0:
+        l_cp = [np.random.randint(0, len(rt_plan_parameters["jaws 1"]), 1)[0]]
+        MU = rt_plan_parameters["weight"][l_cp[0]]
+    nb_part = nb_part / MU
     versa.set_linac_head_motion(
         sim, linac.name, jaws, mlc, rt_plan_parameters, sad=sad, cp_id=l_cp
     )
-    MU = rt_plan_parameters["weight"][l_cp[0]]
-    nb_part = nb_part / MU
+
 
     if sim.visu:
         add_alpha_source(sim, linac.name, z_linac / 2 - 5.6 * mm, 10)