K2 campaign back to string

Author: Nschanche

src/lksearch/K2Search.py

@@ -134,7 +134,7 @@ def _fix_K2_sequence(self):
                 if f"c{row['sequence_number']}{half}" in row["productFilename"]:
                     seq_num[index] = f"{int(row['sequence_number']):02d}{letter}"
 
-        self.table["campaign"] = seq_num.astype(int)
+        self.table["campaign"] = seq_num
 
     def _sort_K2(self):
         # No specific preference for K2 HLSPs

tests/test_missionsearch.py

@@ -235,7 +235,8 @@ def test_properties():
     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()
+    # Campaign is a string, as campaigns can have 'a' and 'b' components
+    assert (result.campaign == "3").all()
 
     result = KeplerSearch("KIC 11904151", exptime="short", quarter=[2, 3, 4]).cubedata
     assert len(result) == 5
@@ -251,7 +252,7 @@ def test_properties():
     assert (result.pipeline == "SPOC").all()
 
     with pytest.raises(AttributeError, match="no attibute"):
-        result = MASTSearch("EPIC 205998445", search_radius=900).cubedata
+        MASTSearch("EPIC 205998445", mission='TESS', search_radius=900).cubedata.sector
 
 
 def test_source_confusion():