From c912a0f733b27c267d1fcc18448f509cb4578d38 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 | 2 ++ 1 file changed, 2 insertions(+) diff --git a/lcviz/tests/test_parser.py b/lcviz/tests/test_parser.py index 4bac62dc..f70691ac 100644 --- a/lcviz/tests/test_parser.py +++ b/lcviz/tests/test_parser.py @@ -5,6 +5,7 @@ from astropy.utils.data import download_file from lightkurve import LightCurve, search_targetpixelfile from lightkurve.io import kepler +from lightkurve.targetpixelfile import TargetPixelFile import astropy.units as u from lcviz.utils import TimeCoordinates @@ -58,6 +59,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=TargetPixelFile).shape == (4447, 4, 6) def test_synthetic_lc(helper):