From 3ea64e05dd9bb08dab610ad1981dd1574856006c Mon Sep 17 00:00:00 2001 From: Tyler Pritchard Date: Fri, 26 Apr 2024 12:13:54 -0400 Subject: [PATCH] ruff --- src/tssc/TESSSearch.py | 4 +++- tests/test_search.py | 8 ++++---- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/src/tssc/TESSSearch.py b/src/tssc/TESSSearch.py index 4cab112..9f7ba5e 100644 --- a/src/tssc/TESSSearch.py +++ b/src/tssc/TESSSearch.py @@ -394,7 +394,9 @@ def search_individual_ffi( new_table.table["year"] = new_table.table["obs_id"].apply( (lambda x: int(x.split("-")[0][4:8])) ) - new_table.table["sector"] = new_table.table["obs_id"].apply(lambda x: int(x.split("-")[1][1:])) + new_table.table["sector"] = new_table.table["obs_id"].apply( + lambda x: int(x.split("-")[1][1:]) + ) return new_table def filter_table( diff --git a/tests/test_search.py b/tests/test_search.py index 471e9c2..5457a5f 100644 --- a/tests/test_search.py +++ b/tests/test_search.py @@ -45,7 +45,9 @@ def test_search_cubedata(): assert len(MASTSearch(tic, mission="TESS").table) > 1 assert ( len( - TESSSearch(tic, pipeline="SPOC", sector=1, search_radius=100).timeseries.table + TESSSearch( + tic, pipeline="SPOC", sector=1, search_radius=100 + ).timeseries.table ) == 2 ) @@ -257,7 +259,6 @@ def test_issue_472(): TESSSearch("TIC41336498", sector=2).tesscut - """ This test used in OG Lightkurve used the fact that we were failing when we tried to read the file into memmory after a download call. In this package we are never reading a file into memmory so we cannot trivially replicate this test. @@ -350,11 +351,10 @@ def test_overlapping_targets_718(): assert len(search) > 1 - def test_tesscut_795(): """Regression test for #795: make sure the __repr__.of a TESSCut SearchResult works.""" - str(TESSSearch("KIC 8462852")) + str(TESSSearch("KIC 8462852")) def test_exptime_filtering():