From bb7473e5c0229d87e04a814e167128747bfc835d Mon Sep 17 00:00:00 2001 From: Kyle Conroy Date: Tue, 23 Jan 2024 12:55:47 -0500 Subject: [PATCH] add test coverage through get_object --- lcviz/tests/test_parser.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lcviz/tests/test_parser.py b/lcviz/tests/test_parser.py index 4bac62dc..9da8f99e 100644 --- a/lcviz/tests/test_parser.py +++ b/lcviz/tests/test_parser.py @@ -3,7 +3,7 @@ from glue.core.roi import XRangeROI, YRangeROI from astropy.time import Time from astropy.utils.data import download_file -from lightkurve import LightCurve, search_targetpixelfile +from lightkurve import LightCurve, KeplerTargetPixelFile, search_targetpixelfile from lightkurve.io import kepler import astropy.units as u @@ -58,6 +58,7 @@ def test_kepler_tpf_via_lightkurve(helper): quarter=10).download() helper.load_data(tpf) assert helper.get_data().shape == (4447, 4, 6) # (time, x, y) + assert helper.app.data_collection[0].get_object(cls=KeplerTargetPixelFile).shape == (447, 4, 6) def test_synthetic_lc(helper):