36 lines
2.0 KiB
Python
36 lines
2.0 KiB
Python
|
from unittest import TestCase
|
||
|
|
from esbo_etc.classes import *
|
||
|
|
import astropy.units as u
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
|
||
|
|
class OpticalComponent(AOpticalComponent):
|
||
|
|
def __init__(self, parent: ITransmissive, transreflectivity: SpectralQty = None,
|
||
|
|
noise: SpectralQty = None, obstruction: float = 0, obstructor_temp: u.Quantity = 0 * u.K,
|
||
|
|
obstructor_emissivity: float = 0):
|
||
|
|
super().__init__(parent, transreflectivity, noise, obstruction, obstructor_temp, obstructor_emissivity)
|
||
|
|
|
||
|
|
|
||
|
|
class TestAOpticalComponent(TestCase):
|
||
|
|
wl = np.arange(4, 8, 1) << u.um
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.target = BlackBodyTarget(self.wl, temp=5778 * u.K, mag=10 * u.mag, band="U")
|
||
|
|
self.comp = OpticalComponent(self.target, SpectralQty(self.wl, [0.5] * 4),
|
||
|
|
SpectralQty(self.wl, [1e-5] * 4 << u.W / (u.m ** 2 * u.nm * u.sr)),
|
||
|
|
obstruction=0.1, obstructor_temp=300 * u.K, obstructor_emissivity=1)
|
||
|
|
|
||
|
|
def test_calcSignal(self):
|
||
|
|
self.assertEqual(self.comp.calcSignal(), SpectralQty(self.wl, [1.25575776e-17, 5.50570557e-18, 2.77637739e-18,
|
||
|
|
1.54664415e-18] << u.W / (u.m ** 2 * u.nm)))
|
||
|
|
|
||
|
|
def test_calcNoise(self):
|
||
|
|
self.assertEqual(self.comp.calcNoise(),
|
||
|
|
SpectralQty(self.wl, [8.21976423e-05, 2.70268340e-04, 5.27503292e-04,
|
||
|
|
7.60597616e-04] << u.W / (u.m ** 2 * u.nm * u.sr)))
|
||
|
|
comp = OpticalComponent(self.comp, SpectralQty(self.wl, [0.5] * 4),
|
||
|
|
SpectralQty(self.wl, [0] * 4 << u.W / (u.m ** 2 * u.nm * u.sr)),
|
||
|
|
obstruction=0.1, obstructor_temp=300 * u.K, obstructor_emissivity=1)
|
||
|
|
self.assertEqual(comp.calcNoise(), SpectralQty(self.wl, [1.09186581e-04, 3.81889092e-04, 7.54879773e-04,
|
||
|
|
10.92866544e-04] << u.W / (u.m ** 2 * u.nm * u.sr)))
|