add test coverage through get_object

spacetelescope · Jan 23, 2024 · bb7473e · bb7473e
1 parent 3e9e016
commit bb7473e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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):