Fix tests

igmhub · Mar 28, 2024 · ac9c0c1 · ac9c0c1
1 parent f5ed793
commit ac9c0c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_astropy_cosmo.py b/tests/test_astropy_cosmo.py
@@ -34,12 +34,12 @@ def test_comoving_distance():
     assert result is not None
     assert isinstance(result, np.ndarray)
     assert len(result) == 3
-    assert float(cosmo.get_dist_c(2.3)) == 10790.582214326228
+    assert np.isclose(float(cosmo.get_dist_c(2.3)), 10790.582214326228)
 
 
 def test_comoving_transverse_distance():
     cosmo = AstropyCosmo(0.3, 0., 7e-5, -1, 67.36, True)
     result = cosmo._comoving_transverse_distance(1.0)
     assert result is not None
     assert isinstance(result, float)
-    assert float(cosmo.get_dist_m(3.4)) == 4698.831363985478
+    assert np.isclose(float(cosmo.get_dist_m(3.4)), 4698.831363985478)