From 08f672caaa62d21c867d5f7b9336d8f58ed8128a Mon Sep 17 00:00:00 2001 From: LukasK13 Date: Wed, 8 Apr 2020 13:38:48 +0200 Subject: [PATCH] Test the target created from a file --- tests/__init__.py | 0 tests/test_FileTarget.py | 21 +++++++++++++++++++++ 2 files changed, 21 insertions(+) create mode 100644 tests/__init__.py create mode 100644 tests/test_FileTarget.py 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)