filter TESScut by exptime (#43)

* filter TESScut by exptime

* ruff formatting

* Spelling

Author: Nschanche

README.rst

@@ -112,14 +112,16 @@ Please include a self-contained example that fully demonstrates your problem or
 
 Changelog:
 ==========
+v1.2.2
+  - Updated the logic surrounding TESScut searches to allow only TESScut results [#32]
+  - filter TESScut results by exposure time keyword [#43]
 v1.2.0
   - Made the catalogsearch functions case-insensitive[#38]
   - Added `CHECK_CACHED_FILE_SIZES` configuration parameter, which on download allows local checking of cached files without a remote file-size check[#34][#38]
   - Refactored `~lksearch.MastSearch.download` function to only call `~astroquery.mast.Observations.download` once 
     - This results in a modest speedup O(N), particularly for files in the local cache when `CHECK_CACHED_FILE_SIZES` is `True`.  
   - Updated tests, documentation for the new features
   - Changed the property cloud_uris(self) to cloud_uri(self) [#42]
-  - Updated the logic surrounding TESScut searches to allow only TESScut results [#32]
 v1.1.1
   - Deprecated TESSSearch.search_sector_ffis due to changes in astroquery functionality
 v1.1.0

src/lksearch/TESSSearch.py

@@ -109,7 +109,8 @@ def __init__(
             self.table = pd.DataFrame(columns=table_keys)
 
         if (pipeline == None) or ("tesscut" in [p.lower() for p in pipeline]):
-            self._add_tesscut_products(sector)
+            self._add_tesscut_products(sector, exptime=exptime)
+
         if len(self.table) == 0:
             raise SearchError(f"No data found for target {self.target}.")
         self._add_TESS_mission_product()
@@ -159,7 +160,11 @@ def _add_TESS_mission_product(self):
         self.table["mission_product"] = mission_product
         self.table["sector"] = self.table["sequence_number"]
 
-    def _add_tesscut_products(self, sector_list: Union[int, list[int]]):
+    def _add_tesscut_products(
+        self,
+        sector_list: Union[int, list[int]],
+        exptime: Union[int, tuple[float]],
+    ):
         """Add tesscut product information to the search table
 
         Parameters
@@ -175,6 +180,8 @@ def _add_tesscut_products(self, sector_list: Union[int, list[int]]):
             self.table = pd.concat([self.table, tesscut_info])
         else:
             self.table = tesscut_info
+        mask = self._mask_by_exptime(exptime=exptime)
+        self.table = self.table[mask]
 
     # FFIs only available when using TESSSearch.
     # Use TESS WCS to just return a table of sectors and dates?

tests/test_missionsearch.py

@@ -466,6 +466,12 @@ def test_tesscut():
     assert len(sr) == 1
     assert len(sr2) == 2
 
+    # No 20-s data was taken so this should be empty
+    # Previously the TESScut data was appended after the exposure time cuts
+    #     so it returned 1800s tesscut data [see issue #41]
+    with pytest.raises(SearchError, match="No data"):
+        TESSSearch("TOI 1161", sector=14, exptime=20)
+
 
 class TestMASTSearchFilter:
     results = MASTSearch("Kepler 16b")