repair test_atmosphere test

LSSTDESC · Aug 26, 2024 · e21056e · e21056e
1 parent a72f6e5
commit e21056e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_simulator.py b/tests/test_simulator.py
@@ -30,9 +30,9 @@ def test_atmosphere():
                        ozone_grid=[400, 400, 1], aerosol_grid=[0.0, 0.1, 2])
     atmospheric_grid = a.compute()
     assert np.sum(atmospheric_grid) > 0
-    assert np.all(np.isclose(a.atmgrid[0, a.index_atm_data:], parameters.LAMBDAS))
-    assert not np.any(np.isclose(a.atmgrid[1, a.index_atm_data:], np.zeros_like(parameters.LAMBDAS), rtol=1e-6))
-    assert a.atmgrid.shape == (3, a.index_atm_data + len(parameters.LAMBDAS))
+    assert np.all(np.isclose(a.atmgrid[0, a.index_atm_data:], a.lambdas))
+    assert not np.any(np.isclose(a.atmgrid[1, a.index_atm_data:], np.zeros_like(a.lambdas), rtol=1e-6))
+    assert a.atmgrid.shape == (3, a.index_atm_data + len(a.lambdas))
     a.save_file(a.image_filename.replace('.fits', '_atmsim.fits'))
     assert os.path.isfile('tests/data/reduc_20170605_028_atmsim.fits')
     a.load_file(a.image_filename.replace('.fits', '_atmsim.fits'))