Test the correct pieces of code

StingraySoftware · Dec 3, 2024 · 219df48 · 219df48
1 parent 443da21
commit 219df48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stingray/tests/test_base.py b/stingray/tests/test_base.py
@@ -1113,7 +1113,7 @@ def test_read_timeseries_by_time_intv_check_bad_gtis(self):
 
     @pytest.mark.parametrize("gti_kind", ["same", "one", "multiple"])
     def test_read_apply_gti_lists(self, gti_kind):
-        reader = FITSTimeseriesReader(self.fname, output_class=DummyStingrayTs)
+        reader = FITSTimeseriesReader(self.fname, output_class=DummyStingrayTs)[:]
         if gti_kind == "same":
             gti_list = [reader.gti]
         elif gti_kind == "one":
@@ -1131,7 +1131,7 @@ def test_read_apply_gti_lists(self, gti_kind):
             assert np.allclose(ev.gti, gti_list[i])
 
     def test_read_apply_gti_lists_ignore_empty(self):
-        reader = FITSTimeseriesReader(self.fname, output_class=DummyStingrayTs)
+        reader = FITSTimeseriesReader(self.fname, output_class=DummyStingrayTs)[:]
         gti_list = [[], [[80000000, 80000512]], [[80000513, 80001024]]]
         evs = list(reader.apply_gti_lists(gti_list))
         assert np.allclose(evs[0].gti, gti_list[1])