diff --git a/tests/optical_component/test_AHotOpticalComponent.py b/tests/optical_component/test_AHotOpticalComponent.py index b7e3ec4..37b9fcd 100644 --- a/tests/optical_component/test_AHotOpticalComponent.py +++ b/tests/optical_component/test_AHotOpticalComponent.py @@ -21,7 +21,7 @@ class TestAHotOpticalComponent(TestCase): wl = np.arange(201, 205, 1) << u.nm def setUp(self): - self.target = FileTarget("data/target/target_demo_1.csv") + self.target = FileTarget("data/target/target_demo_1.csv", self.wl) def test___init__(self): comp = HotOpticalComponent(self.target, SpectralQty(self.wl, np.repeat(0.5, 4) << u.dimensionless_unscaled), diff --git a/tests/optical_component/test_Atmosphere.py b/tests/optical_component/test_Atmosphere.py index 13bc0ba..91bf156 100644 --- a/tests/optical_component/test_Atmosphere.py +++ b/tests/optical_component/test_Atmosphere.py @@ -6,7 +6,7 @@ import astropy.units as u class TestAtmosphere(TestCase): def setUp(self): - self.target = FileTarget("data/target/target_demo_1.csv") + self.target = FileTarget("data/target/target_demo_1.csv", np.arange(200, 208) << u.nm) self.atmosphere = Atmosphere(self.target, "data/atmosphere/atmosphere_transmittance_1.csv", "data/atmosphere/atmosphere_emission_1.csv") diff --git a/tests/optical_component/test_BeamSplitter.py b/tests/optical_component/test_BeamSplitter.py index 4fb5d8a..8b6061c 100644 --- a/tests/optical_component/test_BeamSplitter.py +++ b/tests/optical_component/test_BeamSplitter.py @@ -10,7 +10,7 @@ class TestBeamSplitter(TestCase): wl = np.arange(201, 205, 1) << u.nm def setUp(self): - self.target = FileTarget("data/target/target_demo_1.csv") + self.target = FileTarget("data/target/target_demo_1.csv", self.wl) self.splitter = BeamSplitter(self.target, "data/beamsplitter/beamsplitter_transmittance.csv", 0.5, temp=300 * u.K) diff --git a/tests/optical_component/test_Filter.py b/tests/optical_component/test_Filter.py index 4eed642..ebb5c02 100644 --- a/tests/optical_component/test_Filter.py +++ b/tests/optical_component/test_Filter.py @@ -13,7 +13,7 @@ class TestFilter(TestCase): 0.0, 0.0] << u.W / (u.m ** 2 * u.nm))) def test_fromFile(self): - target = FileTarget("data/target/target_demo_1.csv") + target = FileTarget("data/target/target_demo_1.csv", np.arange(200, 210, 1) << u.nm) filt = Filter.fromFile(target, "data/filter/filter_transmittance.csv") self.assertEqual(filt.calcSignal(), SpectralQty(np.arange(200, 210, 1) << u.nm, [1.10e-15, 1.20e-15, 1.30e-15, 1.40e-15, 1.35e-15, 1.44e-15, diff --git a/tests/optical_component/test_Lens.py b/tests/optical_component/test_Lens.py index 71d12ab..edd5595 100644 --- a/tests/optical_component/test_Lens.py +++ b/tests/optical_component/test_Lens.py @@ -10,7 +10,7 @@ class TestLens(TestCase): wl = np.arange(201, 205, 1) << u.nm def setUp(self): - self.target = FileTarget("data/target/target_demo_1.csv") + self.target = FileTarget("data/target/target_demo_1.csv", self.wl) self.lens = Lens(self.target, "data/lens/lens_transmittance.csv", 0.5, temp=300 * u.K) def test___init__(self): diff --git a/tests/optical_component/test_Mirror.py b/tests/optical_component/test_Mirror.py index 20060d4..5669673 100644 --- a/tests/optical_component/test_Mirror.py +++ b/tests/optical_component/test_Mirror.py @@ -10,7 +10,7 @@ class TestMirror(TestCase): wl = np.arange(201, 205, 1) << u.nm def setUp(self): - self.target = FileTarget("data/target/target_demo_1.csv") + self.target = FileTarget("data/target/target_demo_1.csv", self.wl) self.mirror = Mirror(self.target, "data/mirror/mirror_reflectance.csv", 0.5, temp=300 * u.K) def test___init__(self): diff --git a/tests/optical_component/test_StrayLight.py b/tests/optical_component/test_StrayLight.py index 032b645..2d7f0f2 100644 --- a/tests/optical_component/test_StrayLight.py +++ b/tests/optical_component/test_StrayLight.py @@ -6,7 +6,7 @@ import astropy.units as u class TestStrayLight(TestCase): def setUp(self): - self.target = FileTarget("data/target/target_demo_1.csv") + 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):