Add Sector/Quarter/Campaign properties

Author: Nschanche

src/lksearch/K2Search.py

@@ -90,6 +90,11 @@ def __init__(
             self._fix_K2_sequence()
             self._sort_K2()
 
+    @property
+    def campaign(self):
+        """K2 Observing campaign for each data product found."""
+        return self.table["campaign"].values
+
     @property
     def HLSPs(self):
         """return a MASTSearch object with self.table only containing High Level Science Products"""

src/lksearch/KeplerSearch.py

@@ -95,7 +95,11 @@ def __init__(
             self._sort_Kepler()
             # Can't search mast with quarter/month directly, so filter on that after the fact.
             self.table = self.table[self._filter_kepler(quarter, month)]
-
+    @property
+    def quarter(self):
+        """Kepler Observing quarter for each data product found."""
+        return self.table["quarter"].values
+    
     @property
     def HLSPs(self):
         """return a MASTSearch object with self.table only containing High Level Science Products"""

src/lksearch/TESSSearch.py

@@ -117,6 +117,11 @@ def __init__(
 
         self._sort_TESS()
 
+    @property
+    def sector(self):
+        """TESS Observing sector for each data product found."""
+        return self.table["sector"].values
+    
     @property
     def HLSPs(self):
         """return a MASTSearch object with self.table only containing High Level Science Products"""

tests/test_missionsearch.py

@@ -229,11 +229,32 @@ def test_collections():
 
 
 def test_properties():
+    """Test an assortment of class properties"""
     c = SkyCoord("297.5835 40.98339", unit=(u.deg, u.deg))
     assert_almost_equal(KeplerSearch(c, quarter=6).cubedata.ra[0], 297.5835)
     assert_almost_equal(KeplerSearch(c, quarter=6).cubedata.dec[0], 40.98339)
     assert len(KeplerSearch(c, quarter=6).cubedata.target_name) == 1
 
+    result = K2Search("EPIC 205998445", search_radius=900, campaign=3).cubedata
+    assert len(result) == 4
+    assert len(result.cloud_uri) == 4
+    assert (result.campaign == '3').all()
+    
+    result = KeplerSearch("KIC 11904151",  exptime="short", quarter=[2,3,4]).cubedata
+    assert len(result) == 5
+    assert all([r in [2,3,4] for r in result.quarter])
+
+    result = TESSSearch(tic, pipeline="spoc", sector=1, search_radius=100).timeseries
+    assert len(result) == 2
+    assert len(result.sector) == 2
+    assert (result.campaign == '1').all()
+    assert (result.mission == "TESS").all()
+    assert (result.pipeline == "SPOC").all()
+
+
+    with pytest.raises(AttributeError, match="no attibute"):
+        result = MASTSearch("EPIC 205998445", search_radius=900).cubedata
+
 
 def test_source_confusion():
     # Regression test for issue #148.