Added tests for the lhcquad and lhcdipole fields

Author: akeyken

01_element/lhcdipole.tpl

@@ -0,0 +1,16 @@
+d1: drift, l=1*m, aper1=3*cm;
+b1: sbend, magnetGeometryType="lhcleft", l=1*m, aper1=3*cm, angle=-0.5;
+d2: drift, l=1*m, aper1=3*cm;
+
+l0 : line = (d1,b1,d2);
+
+use, period=l0;
+
+sample, all;
+
+beam, particle="proton",
+      energy=7*TeV,
+      X0={{ X0 }}*cm,
+      Y0={{ Y0 }}*cm;
+
+sampler: samplerplacement, z=5*m, aper1=5*m, aper2=5*m, shape="rectangular", partID={2212};

01_element/lhcquadrupole.tpl

@@ -0,0 +1,16 @@
+d1: drift, l=1*m, aper1=3*cm;
+q1: quadrupole, magnetGeometryType="lhcleft", l=1*m, aper1=3*cm, k1=1;
+d2: drift, l=1*m, aper1=3*cm;
+
+l0 : line = (d1,q1,d2);
+
+use, period=l0;
+
+sample, all;
+
+beam, particle="proton",
+      energy=7*TeV,
+      X0={{ X0 }}*cm,
+      Y0={{ Y0 }}*cm;
+
+sampler: samplerplacement, z=5*m, aper1=5*m, aper2=5*m, shape="rectangular", partID={2212};

01_element/test_lhcdipole.py

@@ -0,0 +1,35 @@
+import pytest
+import pybdsim
+import os
+
+def test() :
+
+    os.chdir(os.path.dirname(__file__))
+    
+    base_name     = "lhcdipole"
+    template_name = base_name+".tpl"
+    gmad_name     = base_name+".gmad"
+    root_name     = base_name+".root"
+
+    # check field in primary aperture,
+    # secondary aperture and outside of the secondary aperture (yoke)
+    listX0 = [0,19.4,19.4]
+    listY0 = [0,0,4]
+    listAssert = [1.9062219,-0.1165236,-0.1521038]
+
+    for i in range(len(listAssert)):
+
+        X0,Y0,assertion = listX0[i],listY0[i],listAssert[i]
+
+        data = {
+            'X0': str(X0),
+            'Y0': str(Y0),
+        }
+
+        pybdsim.Run.RenderGmadJinjaTemplate(template_name,gmad_name,data)
+        pybdsim.Run.Bdsim(gmad_name,base_name,ngenerate=1,seed=1)
+
+        d = pybdsim.DataPandas.BDSIMOutput(root_name)
+        s = d.get_sampler("sampler.")
+
+        assert(s['x'][0] == pytest.approx(assertion))

01_element/test_lhcquadrupole.py

@@ -0,0 +1,35 @@
+import pytest
+import pybdsim
+import os
+
+def test() :
+
+    os.chdir(os.path.dirname(__file__))
+    
+    base_name     = "lhcquadrupole"
+    template_name = base_name+".tpl"
+    gmad_name     = base_name+".gmad"
+    root_name     = base_name+".root"
+
+    # check field along diagonal in primary aperture,
+    # secondary aperture and outside of the secondary aperture (yoke)
+    listX0 = [1,20.4,23.4]
+    listY0 = [1,1,4]
+    listAssert = [-0.0198441,0.1741561,0.2050054]
+
+    for i in range(len(listAssert)):
+
+        X0,Y0,assertion = listX0[i],listY0[i],listAssert[i]
+
+        data = {
+            'X0': str(X0),
+            'Y0': str(Y0),
+        }
+
+        pybdsim.Run.RenderGmadJinjaTemplate(template_name,gmad_name,data)
+        pybdsim.Run.Bdsim(gmad_name,base_name,ngenerate=1,seed=1)
+
+        d = pybdsim.DataPandas.BDSIMOutput(root_name)
+        s = d.get_sampler("sampler.")
+
+        assert(s['x'][0] == pytest.approx(assertion))