Improve unit test for quadrupole element

Author: EZoni

schema/MagneticMultipoleParameters.py

@@ -55,7 +55,7 @@ def validate(cls, values: Dict[str, Any]) -> Dict[str, Any]:
                 msg = " ".join(
                     [
                         f"Invalid magnetic multipole parameter: '{key}'.",
-                        "Magnetic multipole parameters must be of the form 'tiltN', 'BnN', or 'BsN'.",
+                        "Magnetic multipole parameters must be of the form 'tiltN', 'BnN', or 'BsN', where 'N' is an integer.",
                     ]
                 )
                 raise ValueError(msg)

tests/test_schema.py

@@ -65,15 +65,23 @@ def test_DriftElement():
     assert not passed
 
 
-# TODO
 def test_QuadrupoleElement():
     # Create one drift element with custom name and length
     element_name = "quadrupole_element"
     element_length = 1.0
+    element_magnetic_multipole_Bn1 = 1.1
+    element_magnetic_multipole_Bn2 = 1.2
+    element_magnetic_multipole_Bs1 = 2.1
+    element_magnetic_multipole_Bs2 = 2.2
+    element_magnetic_multipole_tilt1 = 3.1
+    element_magnetic_multipole_tilt2 = 3.2
     element_magnetic_multipole = MagneticMultipoleParameters(
-        Bn1=1.1,
-        Bs1=2.2,
-        tilt1=3.3,
+        Bn1=element_magnetic_multipole_Bn1,
+        Bs1=element_magnetic_multipole_Bs1,
+        tilt1=element_magnetic_multipole_tilt1,
+        Bn2=element_magnetic_multipole_Bn2,
+        Bs2=element_magnetic_multipole_Bs2,
+        tilt2=element_magnetic_multipole_tilt2,
     )
     element = QuadrupoleElement(
         name=element_name,
@@ -82,6 +90,12 @@ def test_QuadrupoleElement():
     )
     assert element.name == element_name
     assert element.Length == element_length
+    assert element.MagneticMultipoleP.Bn1 == element_magnetic_multipole_Bn1
+    assert element.MagneticMultipoleP.Bs1 == element_magnetic_multipole_Bs1
+    assert element.MagneticMultipoleP.tilt1 == element_magnetic_multipole_tilt1
+    assert element.MagneticMultipoleP.Bn2 == element_magnetic_multipole_Bn2
+    assert element.MagneticMultipoleP.Bs2 == element_magnetic_multipole_Bs2
+    assert element.MagneticMultipoleP.tilt2 == element_magnetic_multipole_tilt2
     # Serialize the Line object to YAML
     yaml_data = yaml.dump(element.model_dump(), default_flow_style=False)
     print(f"\n{yaml_data}")