28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
from unittest import TestCase
|
|
from esbo_etc.classes.optical_component.AHotOpticalComponent import AHotOpticalComponent
|
|
from esbo_etc.classes.ITransmissive import ITransmissive
|
|
from esbo_etc.classes.SpectralQty import SpectralQty
|
|
from esbo_etc.classes.target.FileTarget import FileTarget
|
|
import astropy.units as u
|
|
import numpy as np
|
|
|
|
|
|
class HotOpticalComponent(AHotOpticalComponent):
|
|
def __init__(self, parent: ITransmissive, emissivity: SpectralQty, wl_bins: u.Quantity, temp: u.Quantity):
|
|
super().__init__(parent, emissivity, wl_bins, temp)
|
|
|
|
def propagate(self, sqty: SpectralQty) -> SpectralQty:
|
|
return sqty
|
|
|
|
|
|
class TestAHotOpticalComponent(TestCase):
|
|
wl = np.arange(201, 205, 1) << u.nm
|
|
|
|
def setUp(self):
|
|
self.target = FileTarget("data/target/target_demo_1.csv")
|
|
self.comp = HotOpticalComponent(self.target, SpectralQty(self.wl, [0.5] * 4), wl_bins=self.wl, temp=300 * u.K)
|
|
|
|
def test___init__(self):
|
|
self.assertEqual(self.comp.calcNoise(), SpectralQty(self.wl, [4.31413931e-96, 1.37122214e-95, 4.30844544e-95,
|
|
1.33846280e-94] << u.W / (u.m**2 * u.nm * u.sr)))
|