def test_binned_fit():
with within_directory(datasets_dir):
start, stop = 0, 50
poly = [1]
arrival_times = np.loadtxt('test_event_data.txt')
evt_list = EventListWithDeadTime(
arrival_times=arrival_times,
measurement=np.zeros_like(arrival_times),
n_channels=1,
start_time=arrival_times[0],
stop_time=arrival_times[-1],
dead_time=np.zeros_like(arrival_times))
evt_list.set_polynomial_fit_interval("%f-%f" % (start + 1, stop - 1),
unbinned=False)
evt_list.set_active_time_intervals("0-1")
results = evt_list.get_poly_info()['coefficients']
assert evt_list.time_intervals == TimeIntervalSet.from_list_of_edges(
[0, 1])
assert evt_list._poly_counts.sum() > 0
evt_list.__repr__()
def test_binned_fit():
with within_directory(datasets_dir):
start, stop = 0, 50
poly = [1]
arrival_times = np.loadtxt('test_event_data.txt')
evt_list = EventListWithDeadTime(arrival_times=arrival_times,
measurement=np.zeros_like(arrival_times),
n_channels=1,
start_time=arrival_times[0],
stop_time=arrival_times[-1],
dead_time=np.zeros_like(arrival_times)
)
evt_list.set_polynomial_fit_interval("%f-%f" % (start + 1, stop - 1), unbinned=False)
evt_list.set_active_time_intervals("0-1")
results = evt_list.get_poly_info()['coefficients']
assert evt_list.time_intervals == TimeIntervalSet.from_list_of_edges([0,1])
assert evt_list._poly_counts.sum() > 0
evt_list.__repr__()
def test_unbinned_fit(event_time_series):
start, stop = 0, 50
poly = [1]
arrival_times = event_time_series
evt_list = EventListWithDeadTime(
arrival_times=arrival_times,
measurement=np.zeros_like(arrival_times),
n_channels=1,
start_time=arrival_times[0],
stop_time=arrival_times[-1],
dead_time=np.zeros_like(arrival_times),
)
evt_list.set_polynomial_fit_interval("%f-%f" % (start + 1, stop - 1),
unbinned=True)
results = evt_list.get_poly_info()["coefficients"]
evt_list.set_active_time_intervals("0-1")
assert evt_list.time_intervals == TimeIntervalSet.from_list_of_edges(
[0, 1])
assert evt_list._poly_counts.sum() > 0
evt_list.__repr__()
def test_time_interval_constructor_set():
t1 = TimeInterval(-10.0, 20.0)
t2 = TimeInterval(10.0, 30.0)
ts = TimeIntervalSet([t1, t2])
assert ts[0] == t1
assert ts[1] == t2
# Use strings
ts2 = TimeIntervalSet.from_strings("-10 - -5", "10 - 20", "20-30",
"-10--5")
assert ts2[0].start_time == -10
assert ts2[0].stop_time == -5
assert ts2[-1].start_time == -10
assert ts2[-1].stop_time == -5
assert ts2[1].start_time == 10
assert ts2[1].stop_time == 20
assert ts2[2].start_time == 20
assert ts2[2].stop_time == 30
# Use edges
ts3 = TimeIntervalSet.from_list_of_edges([-2, -1, 0, 1, 2])
assert ts3[0].start_time == -2
assert ts3[0].stop_time == -1
assert ts3[-1].start_time == 1
assert ts3[-1].stop_time == 2
assert ts3[1].start_time == -1
assert ts3[1].stop_time == 0
assert ts3[2].start_time == 0
assert ts3[2].stop_time == 1
# Use start and stops
ts5 = TimeIntervalSet.from_starts_and_stops([-2, -1, 0, 1], [-1, 0, 1, 2])
assert ts5[0].start_time == -2
assert ts5[0].stop_time == -1
assert ts5[-1].start_time == 1
assert ts5[-1].stop_time == 2
assert ts5[1].start_time == -1
assert ts5[1].stop_time == 0
assert ts5[2].start_time == 0
assert ts5[2].stop_time == 1
with pytest.raises(AssertionError):
ts6 = TimeIntervalSet.from_starts_and_stops([-2, -1, 0, 1], [-1, 0, 1])
# test display
ts5.display()
def test_time_interval_constructor_set():
t1 = TimeInterval(-10.0, 20.0)
t2 = TimeInterval(10.0, 30.0)
ts = TimeIntervalSet([t1, t2])
assert ts[0] == t1
assert ts[1] == t2
# Use strings
ts2 = TimeIntervalSet.from_strings("-10 - -5", "10 - 20", "20-30","-10--5")
assert ts2[0].start_time == -10
assert ts2[0].stop_time == -5
assert ts2[-1].start_time == -10
assert ts2[-1].stop_time == -5
assert ts2[1].start_time == 10
assert ts2[1].stop_time == 20
assert ts2[2].start_time == 20
assert ts2[2].stop_time == 30
# Use edges
ts3 = TimeIntervalSet.from_list_of_edges([-2,-1,0,1,2])
assert ts3[0].start_time == -2
assert ts3[0].stop_time == -1
assert ts3[-1].start_time == 1
assert ts3[-1].stop_time == 2
assert ts3[1].start_time == -1
assert ts3[1].stop_time == 0
assert ts3[2].start_time == 0
assert ts3[2].stop_time == 1
# Use start and stops
ts5 = TimeIntervalSet.from_starts_and_stops([-2, -1, 0, 1], [-1, 0, 1, 2])
assert ts5[0].start_time == -2
assert ts5[0].stop_time == -1
assert ts5[-1].start_time == 1
assert ts5[-1].stop_time == 2
assert ts5[1].start_time == -1
assert ts5[1].stop_time == 0
assert ts5[2].start_time == 0
assert ts5[2].stop_time == 1
with pytest.raises(AssertionError):
ts6 = TimeIntervalSet.from_starts_and_stops([-2, -1, 0, 1], [-1, 0, 1])
# test display
ts5.display()
