def test_vollath_f4(data_dir):
data = fits.getdata(os.path.join(data_dir, 'unsolved.fits'))
data = focus_utils.mask_saturated(data)
assert focus_utils.vollath_F4(data) == pytest.approx(14667.207897717599)
assert focus_utils.vollath_F4(
data, axis='Y') == pytest.approx(14380.343807477504)
assert focus_utils.vollath_F4(
data, axis='X') == pytest.approx(14954.071987957694)
with pytest.raises(ValueError):
focus_utils.vollath_F4(data, axis='Z')
def test_focus_metric_bad_string(data_dir):
data = fits.getdata(os.path.join(data_dir, 'unsolved.fits'))
data = focus_utils.mask_saturated(data)
with pytest.raises(KeyError):
focus_utils.focus_metric(data, merit_function='NOTAMERITFUNCTION')
def _make_pretty_from_fits(fname=None,
title=None,
figsize=(10, 10 / 1.325),
dpi=150,
alpha=0.2,
number_ticks=7,
clip_percent=99.9,
**kwargs):
with open_fits(fname) as hdu:
header = hdu[0].header
data = hdu[0].data
data = focus_utils.mask_saturated(data)
wcs = WCS(header)
if not title:
field = header.get('FIELD', 'Unknown field')
exptime = header.get('EXPTIME', 'Unknown exptime')
filter_type = header.get('FILTER', 'Unknown filter')
try:
date_time = header['DATE-OBS']
except KeyError:
# If we don't have DATE-OBS, check filename for date
try:
basename = os.path.splitext(os.path.basename(fname))[0]
date_time = date_parser.parse(basename).isoformat()
except Exception:
# Otherwise use now
date_time = current_time(pretty=True)
date_time = date_time.replace('T', ' ', 1)
title = '{} ({}s {}) {}'.format(field, exptime, filter_type, date_time)
norm = ImageNormalize(interval=PercentileInterval(clip_percent), stretch=LogStretch())
fig = Figure()
FigureCanvas(fig)
fig.set_size_inches(*figsize)
fig.dpi = dpi
if wcs.is_celestial:
ax = fig.add_subplot(1, 1, 1, projection=wcs)
ax.coords.grid(True, color='white', ls='-', alpha=alpha)
ra_axis = ax.coords['ra']
ra_axis.set_axislabel('Right Ascension')
ra_axis.set_major_formatter('hh:mm')
ra_axis.set_ticks(
number=number_ticks,
color='white',
exclude_overlapping=True
)
dec_axis = ax.coords['dec']
dec_axis.set_axislabel('Declination')
dec_axis.set_major_formatter('dd:mm')
dec_axis.set_ticks(
number=number_ticks,
color='white',
exclude_overlapping=True
)
else:
ax = fig.add_subplot(111)
ax.grid(True, color='white', ls='-', alpha=alpha)
ax.set_xlabel('X / pixels')
ax.set_ylabel('Y / pixels')
im = ax.imshow(data, norm=norm, cmap=palette, origin='lower')
fig.colorbar(im)
fig.suptitle(title)
new_filename = fname.replace('.fits', '.jpg')
fig.savefig(new_filename, bbox_inches='tight')
# explicitly close and delete figure
fig.clf()
del fig
return new_filename