def test_split_lc_by_gtis_minpoints(self): times = [1, 2, 3, 4, 5, 6, 7, 8] counts = [1, 1, 1, 1, 2, 3, 3, 2] gti = [[0.5, 3.5], [3.5, 5.5], [5.5, 8.5]] min_points = 3 lc = Lightcurve(times, counts, gti=gti) list_of_lcs = lc.split_by_gti(min_points=min_points) lc0 = list_of_lcs[0] lc1 = list_of_lcs[1] assert np.all(lc0.time == [1, 2, 3]) assert np.all(lc1.time == [6, 7, 8]) assert np.all(lc0.counts == [1, 1, 1]) assert np.all(lc1.counts == [3, 3, 2])
def test_split_lc_by_gtis(self): times = [1, 2, 3, 4, 5, 6, 7, 8] counts = [1, 1, 1, 1, 2, 3, 3, 2] gti = [[0.5, 4.5], [5.5, 7.5]] lc = Lightcurve(times, counts, gti=gti) list_of_lcs = lc.split_by_gti() lc0 = list_of_lcs[0] lc1 = list_of_lcs[1] assert np.all(lc0.time == [1, 2, 3, 4]) assert np.all(lc1.time == [6, 7]) assert np.all(lc0.counts == [1, 1, 1, 1]) assert np.all(lc1.counts == [3, 3]) assert np.all(lc0.gti == [[0.5, 4.5]]) assert np.all(lc1.gti == [[5.5, 7.5]])