24 lines
1.3 KiB
Python
24 lines
1.3 KiB
Python
from unittest import TestCase
|
|
from esbo_etc import StrayLight, FileTarget, SpectralQty
|
|
import numpy as np
|
|
import astropy.units as u
|
|
|
|
|
|
class TestStrayLight(TestCase):
|
|
def setUp(self):
|
|
self.target = FileTarget("data/target/target_demo_1.csv", np.arange(200, 210) << u.nm)
|
|
self.zodiac = StrayLight(self.target, "data/straylight/zodiacal_emission_1.csv")
|
|
|
|
def test_calcSignal(self):
|
|
self.assertEqual(self.zodiac.calcSignal()[0],
|
|
SpectralQty(np.arange(200, 210) << u.nm, np.array([1.1e-15, 1.2e-15, 1.3e-15, 1.4e-15, 1.5e-15,
|
|
1.6e-15, 1.7e-15, 1.8e-15, 1.9e-15,
|
|
2.0e-15]) << u.W / (u.m ** 2 * u.nm)))
|
|
|
|
def test_calcBackground(self):
|
|
self.assertEqual(self.zodiac.calcBackground(),
|
|
SpectralQty(np.arange(200, 210) << u.nm, np.array([1.1e-16, 1.2e-16, 1.3e-16, 1.4e-16, 1.5e-16,
|
|
1.6e-16, 1.7e-16, 1.8e-16, 1.9e-16,
|
|
2.0e-16]) << u.W /
|
|
(u.m ** 2 * u.nm * u.sr)))
|