test: Handle float comparison with Pytest

scarrazza · Mar 4, 2024 · f5c0371 · f5c0371
1 parent 64768af
commit f5c0371
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/tests/__pycache__/test_basics.cpython-311-pytest-8.0.2.pyc b/tests/__pycache__/test_basics.cpython-311-pytest-8.0.2.pyc
diff --git a/tests/test_basics.py b/tests/test_basics.py
@@ -1,20 +1,21 @@
 import fiatlux
+import pytest
 
 
 def a(q):
     return 0.1
 
 
 def f2(x, q):
-    return 0.
+    return 0.0
 
 
 def fl(x, q):
-    return 0.
+    return 0.0
 
 
 def f2lo(x, q):
-    return 0.
+    return 0.0
 
 
 def test_loading():
@@ -23,7 +24,7 @@ def test_loading():
     lux.PlugStructureFunctions(f2, fl, f2lo)
     lux.InsertInelasticSplitQ([4.18, 1e100])
     pht = lux.EvaluatePhoton(0.1, 1000)
-    assert pht.elastic == 0.0601621248523242
-    assert pht.inelastic_pf == 0.01927555943602428
-    assert pht.msbar_pf == 0.0
-    assert pht.total == 0.07943768428834848
+    assert pytest.approx(pht.elastic) == 0.0601621248523242
+    assert pytest.approx(pht.inelastic_pf) == 0.01927555943602428
+    assert pytest.approx(pht.msbar_pf) == 0.0
+    assert pytest.approx(pht.total) == 0.07943768428834848