def test_build_flux_wl_array_returns_correct_wl_array(self): expected = np.array([]) for i, filter_band in enumerate(self.key): flux, effective_wl = mag2flux(filter_band, self.magnitudes[i]) expected = np.append(expected, effective_wl) expected = u.Quantity(expected, effective_wl.unit) result_wl_array, result_flux_array = build_flux_wl_array(self.key, self.magnitudes) self.assertEqual(expected.value.tolist(), result_wl_array.value.tolist())
def test_mag2flux_returns_correct_effective_wl(self): expected = self.effective_wl result_flux, result_wl = mag2flux(self.filter_band, self.magnitude) self.assertEqual(expected.value, result_wl.value)
def test_flux_at_mag_zero(self): mag = 0.0 expected = self.flux_at_zero_mag result_flux, result_wl = mag2flux(self.filter_band, 0.0) self.assertEqual(expected.value, result_flux.value)
def test_mag2flux_converts_mag_to_correct_flux(self): expected = self.flux_at_zero_mag * 10**(-0.4 * self.magnitude) result_flux, result_wl = mag2flux(self.filter_band, self.magnitude) self.assertEqual(expected.value, result_flux.value)