[TASK] added tests checking calling signature of digital twin

Functional tests pending

python/tests/closed_orbit_test.py

@@ -14,7 +14,13 @@
 import gtpsa
 from thor_scsi.factory import accelerator_from_config
 from thor_scsi.utils.closed_orbit import compute_closed_orbit
+from thor_scsi.utils import canonical_variables_mapping_default
 
+from thor_scsi.utils.accelerator import (
+ instrument_with_standard_observers,
+ extract_orbit_from_standard_observers,
+ extract_orbit_from_accelerator_with_standard_observers,
+)
 
 path = (
  Path(os.path.dirname(__file__))
@@ -28,6 +34,7 @@ def create_acc():
 
 
 def test10_closed_orbit_for_zero():
+
  calc_config = tslib.ConfigType()
  acc = create_acc()
 
@@ -56,6 +63,20 @@ def test10_closed_orbit_for_zero():
  assert x0.ct == pytest.approx(ct_save, abs=1e-9)
 
 
+def test20_closed_orbit_for_zero_ovservers():
+ """test functions used by twin
+
+ just that they execute without error
+ """
+ calc_config = tslib.ConfigType()
+ acc = create_acc()
+ observers_non_perturbated = instrument_with_standard_observers(
+ acc, mapping=canonical_variables_mapping_default
+ )
+ result = compute_closed_orbit(acc, calc_config, delta=0e0)
+ assert result.found_closed_orbit == True
+ extract_orbit_from_accelerator_with_standard_observers(acc)
+
 def compensate_quadrupole_offset_with_dipole_kick(quad):
  dx = quad.get_dx()
  assert quad.get_main_multipole_number() == 2

python/tests/twiss_test.py

@@ -0,0 +1,51 @@
+from thor_scsi.utils import canonical_variables_mapping_default
+from thor_scsi.utils.linear_optics import compute_Twiss_along_lattice
+from thor_scsi.factory import accelerator_from_config
+from thor_scsi import lib as tslib
+import gtpsa
+import pytest
+from pathlib import Path
+import os.path
+
+desc = gtpsa.desc(6, 1)
+
+path = (
+ Path(os.path.dirname(__file__))
+ / "lattices"
+ / "b2_stduser_beamports_blm_tracy_corr.lat"
+)
+
+
+def create_acc():
+ acc = accelerator_from_config(path)
+ acc.set_log_level(tslib.accelerator_log_level.warning)
+ return acc
+
+
+def test_twiss_without_radiation():
+ """Functions as used by digital twin"""
+ n_dof = 2
+ calc_config = tslib.ConfigType()
+ calc_config.radiation = False
+ calc_config.Cavity_on = False
+ acc = create_acc()
+ compute_Twiss_along_lattice(
+ n_dof, acc, calc_config, desc=desc, mapping=canonical_variables_mapping_default
+ )
+
+
+@pytest.mark.skip
+def test_twiss_with_radiation():
+ """Functions as used by digital twin
+
+ just check that it runs
+ """
+ calc_config = tslib.ConfigType()
+ n_dof = 3
+ calc_config.radiation = True
+ calc_config.Cavity_on = True
+ calc_config.Energy = 2.7
+ acc = create_acc()
+ compute_Twiss_along_lattice(
+ n_dof, acc, calc_config, desc=desc, mapping=canonical_variables_mapping_default
+ )