def test_resample_only(input_ad_list, caplog):
p = GMOSLongslit(input_ad_list)
p.resampleToCommonFrame(force_linear=False)
_check_params(caplog.records, 'w1=508.198 w2=1088.323 dw=0.151 npix=3841')
caplog.clear()
adout = p.resampleToCommonFrame(dw=0.15)
assert 'ALIGN' in adout[0].phu
_check_params(caplog.records, 'w1=508.198 w2=1088.232 dw=0.150 npix=3868')
def test_resample_only_and_trim(input_ad_list, caplog):
p = GMOSLongslit(input_ad_list)
# Shouldn't change the first adinput
p.resampleToCommonFrame(trim_data=True, force_linear=False)
_check_params(caplog.records, 'w1=508.198 w2=978.802 dw=0.150 npix=3133')
caplog.clear()
adout = p.resampleToCommonFrame(dw=0.15)
assert 'ALIGN' in adout[0].phu
_check_params(caplog.records, 'w1=508.198 w2=978.802 dw=0.150 npix=3139')
def test_correlation_and_w1_w2(adinputs, caplog):
add_fake_offset(adinputs, offset=10)
p = GMOSLongslit(adinputs)
adout = p.adjustWCSToReference()
assert adout[1].phu['SLITOFF'] == -10
assert adout[2].phu['SLITOFF'] == -20
p.resampleToCommonFrame(dw=0.15, w1=700, w2=850)
_check_params(caplog.records, 'w1=700.000 w2=850.000 dw=0.150 npix=1001')
adstack = p.stackFrames()
assert adstack[0][0].shape == (512, 1001)
def test_correlation_non_linearize(adinputs, caplog):
add_fake_offset(adinputs, offset=10)
p = GMOSLongslit(adinputs)
adout = p.adjustWCSToReference()
assert adout[1].phu['SLITOFF'] == -10
assert adout[2].phu['SLITOFF'] == -20
p.resampleToCommonFrame(force_linear=False)
_check_params(caplog.records, 'w1=508.198 w2=1088.323 dw=0.151 npix=3841')
caplog.clear()
adout = p.resampleToCommonFrame(dw=0.15)
assert 'ALIGN' in adout[0].phu
_check_params(caplog.records, 'w1=508.198 w2=1088.232 dw=0.150 npix=3868')
adstack = p.stackFrames()
assert adstack[0][0].shape == (512, 3868)
def test_resample_and_linearize(input_ad_list, caplog):
p = GMOSLongslit(input_ad_list)
ads = p.resampleToCommonFrame(dw=0.15)
assert len(ads) == 3
assert {len(ad) for ad in ads} == {1}
assert {ad[0].shape[0] for ad in ads} == {3869}
assert {'ALIGN' in ad[0].phu for ad in ads}
_check_params(caplog.records, 'w1=508.198 w2=1088.323 dw=0.150 npix=3869')
def test_resample_linearize_trim_and_stack(input_ad_list, caplog):
p = GMOSLongslit(input_ad_list)
ads = p.resampleToCommonFrame(dw=0.15, trim_data=True)
assert len(ads) == len(test_datasets)
assert len({ad[0].shape[0] for ad in ads}) == 1
_check_params(caplog.records, 'w1=508.198 w2=978.802 dw=0.150 npix=3139')
adout = p.stackFrames()
assert len(adout) == 1
assert len(adout[0]) == 1
assert adout[0][0].shape[0] == 3139
def test_correlation_and_trim(adinputs, caplog):
add_fake_offset(adinputs, offset=10)
p = GMOSLongslit(adinputs)
adout = p.adjustWCSToReference()
assert adout[1].phu['SLITOFF'] == -10
assert adout[2].phu['SLITOFF'] == -20
p.resampleToCommonFrame(dw=0.15, trim_data=True)
_check_params(caplog.records, 'w1=508.198 w2=978.802 dw=0.150 npix=3139')
ad = p.stackFrames()[0]
assert ad[0].shape == (512, 3139)
caplog.clear()
ad = p.findSourceApertures(max_apertures=1)[0]
assert len(ad[0].APERTURE) == 1
np.testing.assert_allclose(ad[0].APERTURE['c0'], 260.8, atol=0.25)
ad = p.extract1DSpectra()[0]
assert ad[0].shape == (3139, )
def test_correlation(adinputs, caplog):
add_fake_offset(adinputs, offset=10)
p = GMOSLongslit(adinputs)
adout = p.adjustWCSToReference()
assert adout[1].phu['SLITOFF'] == -10
assert adout[2].phu['SLITOFF'] == -20
p.resampleToCommonFrame(dw=0.15)
_check_params(caplog.records, 'w1=508.198 w2=1088.323 dw=0.150 npix=3869')
ad = p.stackFrames()[0]
assert ad[0].shape == (512, 3869)
caplog.clear()
ad = p.findSourceApertures(max_apertures=1)[0]
assert len(ad[0].APERTURE) == 1
#assert caplog.records[3].message == 'Found sources at rows: 260.8'
np.testing.assert_allclose(ad[0].APERTURE['c0'], 260.8, atol=0.25)
ad = p.extract1DSpectra()[0]
assert ad[0].shape == (3869, )
def test_resample_error_with_all(input_ad_list, caplog):
p = GMOSLongslit(input_ad_list)
expected_error = "Maximum 3 of w1, w2, dw, npix must be specified"
with pytest.raises(ValueError, match=expected_error):
p.resampleToCommonFrame(dw=0.15, w1=700, w2=850, npix=1001)
def test_resample_and_linearize_with_npix(input_ad_list, caplog):
p = GMOSLongslit(input_ad_list)
p.resampleToCommonFrame(dw=0.15, w1=700, npix=1001)
_check_params(caplog.records, 'w1=700.000 w2=850.000 dw=0.150 npix=1001')