update tests

Author: PabloCorcho

tests/test_dust.py

@@ -166,7 +166,7 @@ def test_cf00_factor_shape(self):
         self.assertTrue(np.isfinite(f.value).all())
 
         # Young should be more attenuated than old at most wavelengths
-        self.assertTrue(np.median(f[0].value) <= np.median(f[1].value))
+        self.assertLessEqual(np.median(f[0].value), np.median(f[1].value))
 
     def test_cf00_apply_raises_for_multicomponent_factor(self):
         model = dust.CharlotFall00Attenuation()

tests/test_observables.py

@@ -75,6 +75,9 @@ def test_filter(self):
 
         fig, ax = filter.plot(show=False)
 
+        self.assertIsNotNone(ax)
+        self.assertIsNotNone(fig)
+
         f_lambda, f_lambda_err = filter.get_flambda_vegamag(
             self.dummy_flam, spectra_err=0.05 * self.dummy_flam
         )

tests/test_sed.py

@@ -1,7 +1,6 @@
 import unittest
 import numpy as np
 from astropy import units as u
-from astropy import constants
 from pst import dust, SSP, sed, models, galaxy
 from pst.observables import load_photometric_filters