Pytest: SP Tolerances (BLAST-ImpactX#1215)

* Pytest: SP Tolerances

* Update tests/python/test_impactx.py

Author: ax3l

tests/python/test_impactx.py

@@ -11,7 +11,7 @@
 import pytest
 from conftest import basepath
 
-from impactx import ImpactX, distribution, elements
+from impactx import Config, ImpactX, distribution, elements
 
 # FIXME in AMReX via https://github.com/AMReX-Codes/amrex/pull/3727
 # def test_impactx_module():
@@ -26,8 +26,16 @@ def validate_fodo(beam):
     """see examples/fodo/analysis_fodo.py"""
     num_particles = beam.total_number_of_particles()
     assert num_particles == 10000
-    atol = 0.0  # ignored
-    rtol = 2.2 * num_particles**-0.5  # from random sampling of a smooth distribution
+    if Config.precision == "SINGLE":
+        atol = 0.0  # ignored
+        rtol = (
+            2.5 * num_particles**-0.5
+        )  # from random sampling of a smooth distribution
+    else:
+        atol = 0.0  # ignored
+        rtol = (
+            2.2 * num_particles**-0.5
+        )  # from random sampling of a smooth distribution
 
     # in situ calculate the reduced beam characteristics
     rbc = beam.beam_moments()
@@ -160,7 +168,7 @@ def test_impactx_nofile():
 
     # simulate full lattice but keep beam global position
     sim.track_particles()
-    assert np.allclose([ref.s], [7.0])
+    assert ref.s == pytest.approx(7.0)
 
     # finalize simulation
     sim.finalize()

tests/python/test_lattice_select.py

@@ -521,9 +521,9 @@ def test_complex_scenarios():
         quad.k *= 1.1  # Increase strength by 10%
 
     # Verify modifications affected original elements
-    assert lattice[1].k == 1.1  # qf1
-    assert lattice[5].k == -1.1  # qd1
-    assert lattice[7].k == 1.1  # qf2
+    assert lattice[1].k == pytest.approx(1.1)  # qf1
+    assert lattice[5].k == pytest.approx(-1.1)  # qd1
+    assert lattice[7].k == pytest.approx(1.1)  # qf2
 
     # Scenario 3b: Find all quadrupoles and modify their strength (type-based)
     all_quads_type = lattice.select(kind=elements.Quad)
@@ -546,8 +546,8 @@ def test_complex_scenarios():
         bend.ds *= 0.9  # Decrease length by 10%
 
     # Verify modifications
-    assert lattice[3].ds == 0.9  # bend1
-    assert lattice[9].ds == 0.9  # bend2
+    assert lattice[3].ds == pytest.approx(0.9)  # bend1
+    assert lattice[9].ds == pytest.approx(0.9)  # bend2
 
     # Scenario 4b: Find all bends and modify their length (type-based)
     all_bends_type = lattice.select(kind=elements.Sbend)