more robust test

WenkeRen · Jun 30, 2024 · 3cb771f · 3cb771f
1 parent e66ae01
commit 3cb771f
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tests/test_dr7sample.py b/tests/test_dr7sample.py
@@ -286,8 +286,17 @@ def test_dr7(nqsofit=20):
 
                 mask = (xid['plate'] == plate) & (xid['mjd'] == mjd) & (xid['fiberID'] == fiber)
                 print(xid[mask])
+
+                if len(xid[mask]) < 1:
+                    print('No spectrum found.')
+                    return
+
                 sp = SDSS.get_spectra(matches=xid[mask])
                 print(sp)
+
+                if len(sp) < 1:
+                    print('No spectrum found.')
+                    return
 
                 data = sp[0]
 
@@ -303,7 +312,7 @@ def test_dr7(nqsofit=20):
                 q.Fit(param_file_name='qsopar.fits', name=None, qso_type='global', host_type='BC03', save_fig=False, save_result=False)
 
                 # Test with host prior
-                #q.Fit(param_file_name='qsopar.fits', name=None, host_prior=True, qso_type='global', host_type='PCA', save_fig=False, save_result=False)
+                q.Fit(param_file_name='qsopar.fits', name=None, host_prior=True, qso_type='global', host_type='PCA', save_fig=False, save_result=False)
 
                 # Emission line loop
                 for j, line in enumerate(line_calc_names):