Check tunes against transfer matrix

Author: austin-hoover

examples/Diagnostics/Tunes/test_tune.py

@@ -153,4 +153,19 @@
         "action_y",
     ] 
 )
-print(particles.iloc[:, 6:])
+print(particles.iloc[:, 6:])
+
+# Check against tune from transfer matrix
+tune_x_true = lattice_params["fractional tune x"]
+tune_y_true = lattice_params["fractional tune y"]
+tune_x_pred = np.mean(phase_data["tune_x"])
+tune_y_pred = np.mean(phase_data["tune_y"])
+
+tune_x_err = tune_x_pred - tune_x_true
+tune_y_err = tune_y_pred - tune_y_true
+
+print("tune_x_err", tune_x_err)
+print("tune_y_err", tune_y_err)
+
+assert np.abs(tune_x_err) < 1.00e-08
+assert np.abs(tune_y_err) < 1.00e-08

examples/Diagnostics/Tunes/test_tune_4d_uncoupled.py

@@ -156,4 +156,19 @@ def build_norm_matrix_from_twiss_2d(alpha: float, beta: float) -> np.ndarray:
         "action_2",
     ] 
 )
-print(particles.iloc[:, 6:])
+print(particles.iloc[:, 6:])
+
+# Check against tune from transfer matrix
+tune_x_true = lattice_params["fractional tune x"]
+tune_y_true = lattice_params["fractional tune y"]
+tune_x_pred = np.mean(phase_data["tune_x"])
+tune_y_pred = np.mean(phase_data["tune_y"])
+
+tune_x_err = tune_x_pred - tune_x_true
+tune_y_err = tune_y_pred - tune_y_true
+
+print("tune_x_err", tune_x_err)
+print("tune_y_err", tune_y_err)
+
+assert np.abs(tune_x_err) < 1.00e-08
+assert np.abs(tune_y_err) < 1.00e-08