add test coverage through get_object

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