def test_find_apertures_with_fake_data(seeing):
"""
Creates a fake AD object with a gaussian profile in spacial direction with a
fwhm defined by the seeing variable in arcsec. Then add some noise, and
test if p.findAperture can find its position.
"""
astrofaker = pytest.importorskip("astrofaker")
gmos_fake_noise = 4 # adu
gmos_plate_scale = 0.0807 # arcsec . px-1
fwhm_to_stddev = 2 * np.sqrt(2 * np.log(2))
ad = astrofaker.create('GMOS-S', mode='SPECT')
ad.init_default_extensions()
y0 = np.random.randint(low=100, high=ad[0].shape[0] - 100)
fwhm = seeing / gmos_plate_scale
stddev = fwhm / fwhm_to_stddev
model = Gaussian1D(mean=y0, stddev=stddev, amplitude=50)
rows, cols = np.mgrid[:ad.shape[0][0], :ad.shape[0][1]]
for ext in ad:
ext.data = model(rows)
ext.data += np.random.poisson(ext.data)
ext.data += (np.random.random(size=ext.data.shape) -
0.5) * gmos_fake_noise
ext.mask = np.zeros_like(ext.data, dtype=np.uint)
p = GMOSSpect([ad])
_ad = p.findSourceApertures()[0]
print(_ad.info)
def test_find_apertures_with_fake_data(peak_position, peak_value, seeing, astrofaker):
"""
Creates a fake AD object with a gaussian profile in spacial direction with a
fwhm defined by the seeing variable in arcsec. Then add some noise, and
test if p.findAperture can find its position.
"""
np.random.seed(42)
gmos_fake_noise = 4 # adu
gmos_plate_scale = 0.0807 # arcsec . px-1
fwhm_to_stddev = 2 * np.sqrt(2 * np.log(2))
ad = astrofaker.create('GMOS-S', mode='SPECT')
ad.init_default_extensions()
fwhm = seeing / gmos_plate_scale
stddev = fwhm / fwhm_to_stddev
model = Gaussian1D(mean=peak_position, stddev=stddev, amplitude=peak_value)
rows, cols = np.mgrid[:ad.shape[0][0], :ad.shape[0][1]]
for ext in ad:
ext.data = model(rows)
ext.data += np.random.poisson(ext.data)
ext.data += (np.random.random(size=ext.data.shape) - 0.5) * gmos_fake_noise
ext.mask = np.zeros_like(ext.data, dtype=np.uint)
p = GMOSSpect([ad])
_ad = p.findSourceApertures(max_apertures=1)[0]
for _ext in _ad:
# ToDo - Could we improve the primitive to have atol=0.50 or less?
np.testing.assert_allclose(_ext.APERTURE['c0'], peak_position, atol=0.6)
def test_find_apertures_using_standard_star(ad_and_center):
"""
Test that p.findApertures can find apertures in Standard Star (which are
normally bright) observations.
"""
ad, expected_center = ad_and_center
p = GMOSSpect([ad])
_ad = p.findSourceApertures(max_apertures=1).pop()
assert hasattr(ad[0], 'APERTURE')
assert len(ad[0].APERTURE) == 1
np.testing.assert_allclose(ad[0].APERTURE['c0'], expected_center, 3)