add test files

Author: ying2212

tests/test_data.py

@@ -18,10 +18,10 @@ def reference_verifier(t, name, bibcode, doi):
 def test_sources(db):
     # Test to verify existing counts of sources and names
     n_sources = db.query(db.Sources).count()
-    assert n_sources == 3619, f"found {n_sources} sources"
+    assert n_sources == 3633, f"found {n_sources} sources"
 
     n_names = db.query(db.Names).count()
-    assert n_names == 12203, f"found {n_names} names"
+    assert n_names == 12217, f"found {n_names} names"
 
 
 
@@ -450,7 +450,10 @@ def test_suar22_ingest(db):
     [
         ("Roja12", 1),
         ("Burg24", 1),
-        ("Zhan18.2054", 32),
+        ("Zhan17.3040", 5),
+        ("Zhan17", 1),
+        ("Zhan18.1352", 2),
+        ("Zhan18.2054", 46),
         ("Lodi17", 28),
         ("Lupi08", 6),
     ],

tests/test_data_spectra.py

@@ -4,14 +4,14 @@
 
 def test_spectra_count(db):
     n_spectra = db.query(db.Spectra).count()
-    assert n_spectra == 1608, f"found {n_spectra} sources"
+    assert n_spectra == 1728, f"found {n_spectra} sources"
 
 
 @pytest.mark.parametrize(
     ("regime", "n_spectra"),
     [
-        ("optical", 744),
-        ("nir", 637),
+        ("optical", 838),
+        ("nir", 663),
         ("mir", 227),
         ("unknown", 0),
     ],
@@ -27,14 +27,16 @@ def test_spectra_regimes(db, regime, n_spectra):
         ("IRTF", 457),
         ("Keck I", 65),
         ("Keck II", 8),
-        ("Magellan I Baade", 9),
+        ("Magellan I Baade", 12),
         ("Magellan II Clay", 11),
         ("SOAR", 2),
         ("Lick Shane 3m", 1),
         ("HST", 77),
         ("Gemini North", 27),
         ("Gemini South", 34),
-        ("ESO VLT", 62),
+        ("GTC", 67),
+        ("ESO VLT", 114),
+        ("SDSS", 1),
         ("Spitzer", 203),
         ("KPNO 2.1m", 93),
         ("KPNO 4m", 251),
@@ -52,6 +54,11 @@ def test_spectra_telescope(db, telescope, n_spectra):
         ("JWST", "NIRSpec", 25),
         ("JWST", "MIRI", 24),
         ("HST", "WFC3", 77),
+        ("ESO VLT", "XShooter", 88),
+        ("GTC", "OSIRIS", 66),
+        ("Magellan I Baade", "IMACS", 1),
+        ("Magellan I Baade", "FIRE", 11),
+        ("SDSS", "BOSS", 1),
     ],
 )
 def test_spectra_instrument(db, telescope, instrument, n_spectra):
@@ -79,6 +86,10 @@ def test_spectra_instrument(db, telescope, instrument, n_spectra):
         ("Burg10.1142", 46),
         ("Manj20", 20),
         ("Roth24", 34),
+        ("Zhan17.3040", 19),
+        ("Zhan17", 4),
+        ("Zhan18.1352", 14),
+        ("Zhan18.2054", 83),
     ],
 )
 def test_spectra_references(db, ref, n_spectra):