Improve unit test for quadrupole element

Author: EZoni

tests/test_schema.py

@@ -70,10 +70,19 @@ 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 +91,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}")