69 lines
4.2 KiB
Python
69 lines
4.2 KiB
Python
from unittest import TestCase
|
|
from esbo_etc.classes.psf.Airy import Airy
|
|
from esbo_etc.classes.sensor.PixelMask import PixelMask
|
|
import astropy.units as u
|
|
import numpy as np
|
|
|
|
|
|
class TestAiry(TestCase):
|
|
def setUp(self):
|
|
self.airy = Airy(13, 4 * u.um, 0.5 * u.m, 10, 6.5 * u.um)
|
|
|
|
def test_calc_reduced_observation_angle(self):
|
|
# No jitter, unobstructed
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("peak").value, 0.0)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("fwhm").value, 1.028)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("min").value, 2.44)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle(80.).value, 1.7938842051009245)
|
|
|
|
# Jitter, unobstructed
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("peak", 1 * u.arcsec).value, 0.0)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("fwhm", 1 * u.arcsec).value, 1.75)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("min", 1 * u.arcsec).value, 3.375)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle(80., 1 * u.arcsec).value, 3.1)
|
|
|
|
# No jitter, obstructed
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("peak", obstruction=0.04).value, 0.0)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("fwhm", obstruction=0.04).value, 1.006752080603888)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("min", obstruction=0.04).value, 2.33301171875)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle(80., obstruction=0.04).value, 3.1045076425044726)
|
|
|
|
# Jitter, obstructed
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("peak", 1 * u.arcsec, 0.04).value, 0.0)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("fwhm", 1 * u.arcsec, 0.04).value, 1.725)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle("min", 1 * u.arcsec, 0.04).value, 3.075)
|
|
self.assertAlmostEqual(self.airy.calcReducedObservationAngle(80., 1 * u.arcsec, 0.04).value, 3.35)
|
|
|
|
def test_mapToPixelArray(self):
|
|
# No jitter, unobstructed
|
|
reduced_observation_angle = self.airy.calcReducedObservationAngle(80).value
|
|
d_ap = (reduced_observation_angle / (6.5 * u.um / (13.0 * 4 * u.um))).decompose() * u.pix
|
|
mask = PixelMask(np.array([1024, 1024]) << u.pix, 6.5 * u.um, np.array([0.5, 0.5]) << u.pix)
|
|
mask.createPhotometricAperture("circle", d_ap / 2)
|
|
mask = self.airy.mapToPixelMask(mask)
|
|
self.assertAlmostEqual(float(mask.sum()), 0.8173985568945881)
|
|
|
|
# Jitter, unobstructed
|
|
reduced_observation_angle = self.airy.calcReducedObservationAngle(80, 1 * u.arcsec).value
|
|
d_ap = (reduced_observation_angle / (6.5 * u.um / (13.0 * 4 * u.um))).decompose() * u.pix
|
|
mask = PixelMask(np.array([1024, 1024]) << u.pix, 6.5 * u.um, np.array([0.5, 0.5]) << u.pix)
|
|
mask.createPhotometricAperture("circle", d_ap / 2)
|
|
mask = self.airy.mapToPixelMask(mask, 1 * u.arcsec)
|
|
self.assertAlmostEqual(float(mask.sum()), 0.8108919935181225)
|
|
|
|
# No jitter, obstructed
|
|
reduced_observation_angle = self.airy.calcReducedObservationAngle(80, obstruction=0.04).value
|
|
d_ap = (reduced_observation_angle / (6.5 * u.um / (13.0 * 4 * u.um))).decompose() * u.pix
|
|
mask = PixelMask(np.array([1024, 1024]) << u.pix, 6.5 * u.um, np.array([0.5, 0.5]) << u.pix)
|
|
mask.createPhotometricAperture("circle", d_ap / 2)
|
|
mask = self.airy.mapToPixelMask(mask, obstruction=0.04)
|
|
self.assertAlmostEqual(float(mask.sum()), 0.8085985979598022)
|
|
|
|
# Jitter, obstructed
|
|
reduced_observation_angle = self.airy.calcReducedObservationAngle(80, 1 * u.arcsec, 0.04).value
|
|
d_ap = (reduced_observation_angle / (6.5 * u.um / (13.0 * 4 * u.um))).decompose() * u.pix
|
|
mask = PixelMask(np.array([1024, 1024]) << u.pix, 6.5 * u.um, np.array([0.5, 0.5]) << u.pix)
|
|
mask.createPhotometricAperture("circle", d_ap / 2)
|
|
mask = self.airy.mapToPixelMask(mask, 1 * u.arcsec, 0.04)
|
|
self.assertAlmostEqual(float(mask.sum()), 0.808837170286202)
|