Code clean up

This commit is contained in:
Lukas Klass 2020-04-08 17:30:25 +02:00
parent 5ef9300353
commit 73f09eabe9
1 changed files with 3 additions and 4 deletions

View File

@ -9,13 +9,12 @@ class TestFileTarget(TestCase):
def setUp(self):
self.target = FileTarget("data/target/target_demo_1.csv")
class TestInit(TestFileTarget):
def test_signal(self):
signal = SpectralQty(np.arange(200, 210, 1) * u.nm,
signal = SpectralQty(np.arange(200, 210, 1) << u.nm,
np.arange(1.1e-15, 2.0e-15, 1e-16) << u.W / (u.m ** 2 * u.nm))
self.assertTrue(self.target.calcSignal().__eq__(signal))
def test_noise(self):
noise = SpectralQty(list(range(200, 210, 1)) * u.nm, [0] * 10 << u.W / (u.m ** 2 * u.nm * u.sr))
noise = SpectralQty(np.arange(200, 210, 1) << u.nm,
[0.] * 10 << u.W / (u.m ** 2 * u.nm * u.sr))
self.assertEqual(self.target.calcNoise(), noise)