Add patch tests

Author: jgray-19

tests/tests/test-ptc-maps/test-patch-maps.mad

@@ -0,0 +1,121 @@
+-- ../mad test-misc-maps.mad
+-- assume ../madx64 to be present...
+
+local object                                                    in MAD
+local tblcat                                                    in MAD.utility
+local run_test in require("trackvsptc")
+
+-- The setup for the tests ----------------------------------------------------o
+
+local ref_cfg = object "ref" {
+  -- How to run the tests
+  dorun  = true, -- Default: true 
+  dosave = true, -- Default: false
+  doprnt = true, -- Default: false
+  dodbg  = true, -- Default: false
+  doplot = true, -- Default: false
+  
+  -- list of variables in the tests
+  alist = {"model", "energy", "method", "nslice", "x0i"}, --attributes list
+  tol = 1000,
+
+  -- list of values that are used to run test but are not changed throughout
+  order  = 4,        -- 2/4
+  icase  = 56,       -- 56/6
+  debug  = 0,        -- 0/6
+  snm    = -1,
+
+  x0i    = 1..4,       -- 0, 4D, 5D, 6D (see get_mad_str)
+}
+
+local function testRotation()
+  local cfg = ref_cfg "rot" { 
+    elm = "${dir}ROTATION, at=${at}, angle=pi/${_angle};",
+    
+    model  = {1},
+    method = {2},    
+    nslice = {1},
+    energy = {1, 6500},  -- {1, 450, 6500}
+
+    tol = 100,
+
+    at    = {0.05, 0.25, 0.75, 1},
+    _angle= {10, 50, 100, 330, 500, 1000},
+    dir   = {"X", "Y", "S"},
+    alist = tblcat(ref_cfg.alist, {"dir", "at", "_angle"}),
+
+    plot_info = {
+      series = {
+        "'${dir}' == 'X'",
+        "'${dir}' == 'Y'",
+        "'${dir}' == 'S'",
+      },
+      legend = {
+        y1 = "X Rotation",
+        y2 = "Y Rotation",
+        y3 = "S Rotation",
+      },
+      title = "Rotation NG vs PTC",
+      filename = "rotation.png",
+    }
+  }
+  run_test(cfg)
+end
+
+local function testTranslate()
+  local cfg = ref_cfg "translate" {
+    elm = "translation, at=${at}, dx=${dx}, dy=${dy}, ds=${ds};",
+
+    model  = {1},
+    method = {2},
+    nslice = {1},
+    energy = {1, 6500},  -- {1, 450, 6500}
+
+    tol = 100,
+
+    at    = {0.05, 0.25, 0.75, 1},
+    dx    = {0.01, 0.05, 0.1, 0.2},
+    dy    = {0.01, 0.05, 0.1, 0.2},
+    ds    = {0.01, 0.05, 0.1, 0.2},
+    alist = tblcat(ref_cfg.alist, {"at", "dx", "dy", "ds"}),
+
+    plot_info = {
+      title = "Translation NG vs PTC",
+      filename = "translation.png",
+    }
+  }
+  run_test(cfg)
+end
+
+local function testChangeRef() -- FAILS -> ptc does rotation x, y, s not y, -x, s as ng
+  local cfg = ref_cfg "changeref" {
+    elm = "changeref, at=${at}, dx=${dx}, dy=${dy}, ds=${ds}, dtheta=${dtheta}, dphi=${dphi}, dpsi=${dpsi};",
+
+    model  = {1},
+    method = {2},
+    nslice = {1},
+    energy = {1, 6500},  -- {1, 450, 6500}
+
+    tol = 100,
+
+    at    = {0.05, 0.25, 0.75, 1},
+    dx    = {0,-0.01, 0.4},
+    dy    = {0,-0.05, 0.3},
+    ds    = {0, 0.03,-0.2},
+    dtheta= {0, 0.01, 0.05, 0.1, 0.2},
+    dphi  = {0, 0.03, 0.05, 0.2, 0.1},
+    dpsi  = {0, 0.02, 0.05, 0.3, 0.6},
+
+    alist = tblcat(ref_cfg.alist, {"at", "dx", "dy", "ds", "dtheta", "dphi", "dpsi"}),
+    plot_info = {
+      title = "ChangeRef NG vs PTC",
+      filename = "changeref.png",
+    }
+  }
+
+  run_test(cfg)
+end
+
+testRotation()
+testTranslate()
+-- testChangeRef()