def test_split_with_three_segments(self): test_time = np.array([1, 2, 3, 6, 7, 8, 10, 11, 12]) test_counts = np.random.rand(len(test_time)) lc_test = Lightcurve(test_time, test_counts) slc = lc_test.split(1.5) assert len(slc) == 3
def test_split_with_three_segments(self): test_time = np.array([1, 2, 3, 6, 7, 8, 10, 11, 12]) test_counts = np.random.rand(len(test_time)) with warnings.catch_warnings(): warnings.simplefilter("ignore", category=UserWarning) lc_test = Lightcurve(test_time, test_counts) slc = lc_test.split(1.5) assert len(slc) == 3
def test_consecutive_gaps(self): test_time = np.array([1, 2, 3, 6, 9, 10, 11]) test_counts = np.random.rand(len(test_time)) lc_test = Lightcurve(test_time, test_counts) slc = lc_test.split(1.5) assert np.all((slc[0].time == [1, 2, 3])) assert np.all((slc[1].time == [9, 10, 11])) assert np.all((slc[0].counts == test_counts[:3])) assert np.all((slc[1].counts == test_counts[4:]))
def test_split_has_correct_data_points(self): test_time = np.array([1, 2, 3, 6, 7, 8]) test_counts = np.random.rand(len(test_time)) lc_test = Lightcurve(test_time, test_counts) slc = lc_test.split(1.5) assert np.all((slc[0].time == [1, 2, 3])) assert np.all((slc[1].time == [6, 7, 8])) assert np.all((slc[0].counts == test_counts[:3])) assert np.all((slc[1].counts == test_counts[3:]))
def test_split_with_gtis(self): test_time = np.array([1, 2, 3, 6, 7, 8, 10, 11, 12]) test_counts = np.random.rand(len(test_time)) gti = [[0, 4], [9, 13]] lc_test = Lightcurve(test_time, test_counts, gti=gti) slc = lc_test.split(1.5) assert np.all((slc[0].time == [1, 2, 3])) assert np.all((slc[1].time == [10, 11, 12])) assert np.all((slc[0].counts == test_counts[:3])) assert np.all((slc[1].counts == test_counts[6:]))
def test_consecutive_gaps(self): test_time = np.array([1, 2, 3, 6, 9, 10, 11]) test_counts = np.random.rand(len(test_time)) with warnings.catch_warnings(): warnings.simplefilter("ignore", category=UserWarning) lc_test = Lightcurve(test_time, test_counts) slc = lc_test.split(1.5) assert np.allclose(slc[0].time, [1, 2, 3]) assert np.allclose(slc[1].time, [9, 10, 11]) assert np.allclose(slc[0].counts, test_counts[:3]) assert np.allclose(slc[1].counts, test_counts[4:])
def test_split_has_correct_data_points(self): test_time = np.array([1, 2, 3, 6, 7, 8]) test_counts = np.random.rand(len(test_time)) with warnings.catch_warnings(): warnings.simplefilter("ignore", category=UserWarning) lc_test = Lightcurve(test_time, test_counts) slc = lc_test.split(1.5) assert np.allclose(slc[0].time, [1, 2, 3]) assert np.allclose(slc[1].time, [6, 7, 8]) assert np.allclose(slc[0].counts, test_counts[:3]) assert np.allclose(slc[1].counts, test_counts[3:])