From d02d9d6197002531facc692fc5383e412b7fb008 Mon Sep 17 00:00:00 2001 From: LukasK13 Date: Wed, 15 Apr 2020 15:39:25 +0200 Subject: [PATCH] Fix arguments --- tests/test_AHotOpticalComponent.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/test_AHotOpticalComponent.py b/tests/test_AHotOpticalComponent.py index 0156829..193eef4 100644 --- a/tests/test_AHotOpticalComponent.py +++ b/tests/test_AHotOpticalComponent.py @@ -8,8 +8,9 @@ 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 __init__(self, parent: ITransmissive, emissivity: SpectralQty, temp: u.Quantity, obstruction: float = 0, + obstructor_temp: u.Quantity = 0 * u.K, obstructor_emissivity: float = 1): + super().__init__(parent, emissivity, temp, obstruction, obstructor_temp, obstructor_emissivity) def propagate(self, sqty: SpectralQty) -> SpectralQty: return sqty @@ -20,7 +21,7 @@ class TestAHotOpticalComponent(TestCase): 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) + self.comp = HotOpticalComponent(self.target, SpectralQty(self.wl, [0.5] * 4), 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,