Skip to content

Commit c90d2d0

Browse files
committed
Convert spectra tests to AlmostEqual's
1 parent 7b99987 commit c90d2d0

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

neoexchange/photometrics/tests/test_spectraplot.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,7 @@ def setUp(self):
209209
files_to_copy = [self.fitsfile, self.analogfile]
210210

211211
self.tolerance = 1
212+
self.precision = 7
212213

213214
self.test_dir = tempfile.mkdtemp(prefix='tmp_neox_')
214215
for test_file in files_to_copy:
@@ -248,8 +249,8 @@ def test_spectrum_plot(self):
248249
self.assertEqual(expected_label, label)
249250
self.assertGreater(wav_range[1], max(wavelength))
250251
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)
253254

254255
def test_reflectance_plot(self):
255256
expected_label = '455432 -- HD 196164 -- 20190727'
@@ -263,7 +264,7 @@ def test_reflectance_plot(self):
263264
self.assertEqual(expected_label, label)
264265
self.assertGreater(wav_range[1], max(wavelength))
265266
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)
268269

269270

0 commit comments

Comments
 (0)