added tesscut test

lightkurve · Dec 19, 2024 · 5ef7e7b · 5ef7e7b
1 parent a9176f9
commit 5ef7e7b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/lksearch/TESSSearch.py b/src/lksearch/TESSSearch.py
@@ -91,7 +91,7 @@ def __init__(
         pipeline = np.atleast_1d(pipeline)
 
         # Sending 'pipeline' to MASTSearch init causes a SearchError, so handle that separately
-        if "tesscut" in [p.lower() for p in pipeline] or (type(pipeline) is type(None)):
+        if (type(pipeline) is type(None)) or ("tesscut" in [p.lower() for p in pipeline]) :
             add_tesscut = True
             pipeline = np.delete(
                 pipeline, [p.lower() for p in pipeline].index("tesscut")

diff --git a/tests/test_missionsearch.py b/tests/test_missionsearch.py
@@ -440,9 +440,12 @@ def test_FFI_retrieval():
 def test_tesscut():
     """Can we find and download TESS tesscut tpfs"""
     results = TESSSearch("Kepler 16b", hlsp=False, sector=14)
+    results2 = TESSSearch("Kepler 16b", pipeline='TESScut', sector=14)
     assert len(results) == 12
     assert len(results.cubedata) == 2
-    manifest = results.cubedata[1].download()
+    assert len(results.tesscut) == 1
+    assert len(results.tesscut) == len(results2)
+    manifest = results.tesscut.download()
     assert len(manifest) == 1