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')