remote data test via searchtargetpixelfile

spacetelescope · Jan 19, 2024 · 98a2c43 · 98a2c43
1 parent 077e5a3
commit 98a2c43
Showing 1 changed file with 11 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
+from lightkurve import LightCurve, search_targetpixelfile
 from lightkurve.io import kepler
 import astropy.units as u
 
@@ -50,6 +50,16 @@ def test_kepler_via_mast_preparsed(helper):
     assert flux.unit.is_equivalent(u.electron / u.s)
 
 
+@pytest.mark.remote_data
+def test_kepler_tpf_via_lightkurve(helper):
+    tpf = search_targetpixelfile("KIC 001429092",
+                                 mission="Kepler",
+                                 cadence="long",
+                                 quarter=10).download()
+    helper.load_data(tpf)
+    assert helper.get_data().shape == (4447, 22, 11)  # (time, x, y)
+
+
 def test_synthetic_lc(helper):
     time = Time(np.linspace(2460050, 2460060), format='jd')
     flux = np.ones(len(time)) * u.electron / u.s