@@ -209,6 +209,7 @@ def setUp(self):
209
209
files_to_copy = [self .fitsfile , self .analogfile ]
210
210
211
211
self .tolerance = 1
212
+ self .precision = 7
212
213
213
214
self .test_dir = tempfile .mkdtemp (prefix = 'tmp_neox_' )
214
215
for test_file in files_to_copy :
@@ -248,8 +249,8 @@ def test_spectrum_plot(self):
248
249
self .assertEqual (expected_label , label )
249
250
self .assertGreater (wav_range [1 ], max (wavelength ))
250
251
self .assertLess (wav_range [0 ], min (wavelength ))
251
- self .assertEqual (np .mean (flux ), expected_flux_mean )
252
- self .assertEqual (np .mean (error ).value , expected_error_mean )
252
+ self .assertAlmostEqual (np .mean (flux ), expected_flux_mean , self . precision )
253
+ self .assertAlmostEqual (np .mean (error ).value , expected_error_mean , self . precision )
253
254
254
255
def test_reflectance_plot (self ):
255
256
expected_label = '455432 -- HD 196164 -- 20190727'
@@ -263,7 +264,7 @@ def test_reflectance_plot(self):
263
264
self .assertEqual (expected_label , label )
264
265
self .assertGreater (wav_range [1 ], max (wavelength ))
265
266
self .assertLess (wav_range [0 ], min (wavelength ))
266
- self .assertEqual (np .mean (flux ), expected_flux_mean )
267
- self .assertEqual (np .mean (error ), expected_error_mean )
267
+ self .assertAlmostEqual (np .mean (flux ), expected_flux_mean , self . precision )
268
+ self .assertAlmostEqual (np .mean (error ), expected_error_mean , self . precision )
268
269
269
270
0 commit comments