From: Ole Streicher Date: Wed, 19 Apr 2017 10:24:00 +0200 Subject: Use assert_almost_equal instead of assert_equal in test_twod_numpy Closes: #860620 --- spectral_cube/tests/test_dask.py | 2 +- spectral_cube/tests/test_spectral_axis.py | 4 ++-- spectral_cube/tests/test_spectral_cube.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/spectral_cube/tests/test_dask.py b/spectral_cube/tests/test_dask.py index 1edb61b..6800fef 100644 --- a/spectral_cube/tests/test_dask.py +++ b/spectral_cube/tests/test_dask.py @@ -112,7 +112,7 @@ def test_statistics_withnans(data_adv): cube.rechunk((1,2,2)) stats = cube.statistics() for key in ('min', 'max', 'sum'): - assert stats[key] == getattr(cube, key)() + assert_allclose(stats[key], getattr(cube, key)()) @pytest.mark.skipif(not CASA_INSTALLED, reason='Requires CASA to be installed') diff --git a/spectral_cube/tests/test_spectral_axis.py b/spectral_cube/tests/test_spectral_axis.py index 5c741b7..26f66ed 100644 --- a/spectral_cube/tests/test_spectral_axis.py +++ b/spectral_cube/tests/test_spectral_axis.py @@ -22,13 +22,13 @@ def test_cube_wcs_freqtovel(): newwcs = convert_spectral_axis(w1, 'km/s', 'VRAD', rest_value=w1.wcs.restfrq*u.Hz) assert newwcs.wcs.ctype[2] == 'VRAD' - assert newwcs.wcs.crval[2] == 305.2461585938794 + np.testing.assert_almost_equal(newwcs.wcs.crval[2], 305.2461585938794) assert newwcs.wcs.cunit[2] == u.Unit('km/s') newwcs = convert_spectral_axis(w1, 'km/s', 'VRAD') assert newwcs.wcs.ctype[2] == 'VRAD' - assert newwcs.wcs.crval[2] == 305.2461585938794 + np.testing.assert_almost_equal(newwcs.wcs.crval[2], 305.2461585938794) assert newwcs.wcs.cunit[2] == u.Unit('km/s') diff --git a/spectral_cube/tests/test_spectral_cube.py b/spectral_cube/tests/test_spectral_cube.py index 4e73ecc..7b67438 100644 --- a/spectral_cube/tests/test_spectral_cube.py +++ b/spectral_cube/tests/test_spectral_cube.py @@ -1333,7 +1333,7 @@ def test_twod_numpy(func, how, axis, filename, use_dask): # data has a redundant 1st axis dproj = getattr(data,func)(axis=(0,axis+1)).squeeze() assert isinstance(proj, Projection) - np.testing.assert_equal(proj.value, dproj) + np.testing.assert_almost_equal(proj.value, dproj) assert cube.unit == proj.unit @pytest.mark.parametrize(('func','how','axis','filename'),