diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/test_FileTarget.py b/tests/test_FileTarget.py new file mode 100644 index 0000000..3ae8dcd --- /dev/null +++ b/tests/test_FileTarget.py @@ -0,0 +1,21 @@ +from unittest import TestCase +from esbo_etc.classes.target.FileTarget import FileTarget +from esbo_etc.classes.SpectralQty import SpectralQty +import astropy.units as u +import numpy as np + + +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, + 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)) + self.assertEqual(self.target.calcNoise(), noise)