add tolerance to allclose

rice-solar-physics · May 21, 2024 · 16a81bf · 16a81bf
1 parent 29bc38b
commit 16a81bf
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pydrad/parse/tests/test_strand.py b/pydrad/parse/tests/test_strand.py
@@ -97,11 +97,13 @@ def test_term_file_output(strand):
     for p in strand:
         # The electron energy equation's numerical viscosity term is always 0:
         assert u.allclose(p.electron_numerical_viscosity,
-                        np.zeros_like(p.electron_numerical_viscosity)
+                        np.zeros_like(p.electron_numerical_viscosity),
+                        rtol=0.0, atol=1e-8*u.erg/u.s/u.cm**3,
                         )
         # The hydrogen energy equation's gravity term is never 0:
         assert not u.allclose(p.hydrogen_gravity,
-                            np.zeros_like(p.hydrogen_gravity)
+                            np.zeros_like(p.hydrogen_gravity),
+                            rtol=0.0, atol=1e-8*u.erg/u.s/u.cm**3,
                         )
 
 def test_term_file_units(strand):