Add utility functions and remove blank lines

MMannieCorbisier · MMannieCorbisier · commit 73e5255f3549 · 2025-12-01T16:47:03.000Z
diff --git a/01_element/test_drift2.py b/01_element/test_drift2.py
@@ -72,7 +72,6 @@ def sim_env():
     
     
 class TestClass:
-
     @pytest.mark.parametrize(
     "Ek_in, p_type",
     [
@@ -91,7 +90,6 @@ def test_drift2(self, sim_env, Ek_in, p_type) :
         }      
         pybdsim.Run.RenderGmadJinjaTemplate(sim_env["template_name"],sim_env["gmad_name"],data)
         pybdsim.Run.Bdsim(sim_env["gmad_name"],sim_env["base_name"],sim_env["n_in"],1)
-
         with uproot.open(sim_env["root_name"]) as file:
             samplerName="d1."
             f=file['Event'][samplerName]

Original file line number	Diff line number	Diff line change
`@@ -72,7 +72,6 @@ def sim_env():`
`72`	`72`
`73`	`73`
`74`	`74`	`class TestClass:`
`75`		`-`
`76`	`75`	`@pytest.mark.parametrize(`
`77`	`76`	`"Ek_in, p_type",`
`78`	`77`	`[`
`@@ -91,7 +90,6 @@ def test_drift2(self, sim_env, Ek_in, p_type) :`
`91`	`90`	`}`
`92`	`91`	`pybdsim.Run.RenderGmadJinjaTemplate(sim_env["template_name"],sim_env["gmad_name"],data)`
`93`	`92`	`pybdsim.Run.Bdsim(sim_env["gmad_name"],sim_env["base_name"],sim_env["n_in"],1)`
`94`		`-`
`95`	`93`	`with uproot.open(sim_env["root_name"]) as file:`
`96`	`94`	`samplerName="d1."`
`97`	`95`	`f=file['Event'][samplerName]`