def test_read_gbm_tte():
with within_directory(datasets_directory):
data_dir = os.path.join('gbm', 'bn080916009')
nai3 = TimeSeriesBuilder.from_gbm_tte(
'NAI3',
os.path.join(data_dir, "glg_tte_n3_bn080916009_v01.fit.gz"),
rsp_file=os.path.join(data_dir,
"glg_cspec_n3_bn080916009_v00.rsp2"),
poly_order=-1)
nai3.set_active_time_interval('0-1')
nai3.set_background_interval('-20--10', '100-200')
speclike = nai3.to_spectrumlike()
assert isinstance(speclike, DispersionSpectrumLike)
assert not speclike.background_spectrum.is_poisson
speclike = nai3.to_spectrumlike(extract_measured_background=True)
assert isinstance(speclike, DispersionSpectrumLike)
assert speclike.background_spectrum.is_poisson
# test binning
# should not have bins yet
with pytest.raises(RuntimeError):
nai3.bins
# First catch the errors
# This is without specifying the correct options name
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method='constant')
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method='significance')
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method='constant', p0=.1)
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0,
stop=10,
method='significance',
dt=1)
# now incorrect options
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method='not_a_method')
# Now test values
nai3.create_time_bins(start=0, stop=10, method='constant', dt=1)
assert len(nai3.bins) == 10
assert nai3.bins.argsort() == range(len(nai3.bins))
nai3.create_time_bins(start=0, stop=10, method='bayesblocks', p0=.1)
assert nai3.bins.argsort() == range(len(nai3.bins))
assert len(nai3.bins) == 5
nai3.create_time_bins(start=0,
stop=10,
method='significance',
sigma=40)
assert nai3.bins.argsort() == range(len(nai3.bins))
assert len(nai3.bins) == 5
nai3.view_lightcurve(use_binner=True)
nai3.write_pha_from_binner('test_from_nai3', overwrite=True)
def test_read_gbm_tte():
with within_directory(datasets_directory):
data_dir = os.path.join("gbm", "bn080916009")
nai3 = TimeSeriesBuilder.from_gbm_tte(
"NAI3",
os.path.join(data_dir, "glg_tte_n3_bn080916009_v01.fit.gz"),
rsp_file=os.path.join(data_dir,
"glg_cspec_n3_bn080916009_v00.rsp2"),
poly_order=-1,
)
nai3.set_active_time_interval("0-1")
nai3.set_background_interval("-20--10", "100-200", unbinned=False)
speclike = nai3.to_spectrumlike()
assert isinstance(speclike, DispersionSpectrumLike)
assert not speclike.background_spectrum.is_poisson
speclike = nai3.to_spectrumlike(extract_measured_background=True)
assert isinstance(speclike, DispersionSpectrumLike)
assert speclike.background_spectrum.is_poisson
# test binning
# should not have bins yet
with pytest.raises(RuntimeError):
nai3.bins
# First catch the errors
# This is without specifying the correct options name
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method="constant")
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method="significance")
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method="constant", p0=0.1)
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0,
stop=10,
method="significance",
dt=1)
# now incorrect options
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method="not_a_method")
# Now test values
nai3.create_time_bins(start=0, stop=10, method="constant", dt=1)
assert len(nai3.bins) == 10
assert nai3.bins.argsort() == list(range(len(nai3.bins)))
nai3.create_time_bins(start=0, stop=10, method="bayesblocks", p0=0.1)
assert nai3.bins.argsort() == list(range(len(nai3.bins)))
assert len(nai3.bins) == 5
nai3.create_time_bins(start=0,
stop=10,
method="significance",
sigma=40)
assert nai3.bins.argsort() == list(range(len(nai3.bins)))
assert len(nai3.bins) == 5
nai3.view_lightcurve(use_binner=True)
nai3.write_pha_from_binner("test_from_nai3",
overwrite=True,
force_rsp_write=True)
def test_read_gbm_tte():
with within_directory(datasets_directory):
data_dir = os.path.join('gbm', 'bn080916009')
nai3 = TimeSeriesBuilder.from_gbm_tte('NAI3',
os.path.join(data_dir, "glg_tte_n3_bn080916009_v01.fit.gz"),
rsp_file=os.path.join(data_dir, "glg_cspec_n3_bn080916009_v00.rsp2"),
poly_order=-1)
nai3.set_active_time_interval('0-1')
nai3.set_background_interval('-20--10', '100-200')
speclike = nai3.to_spectrumlike()
assert isinstance(speclike, DispersionSpectrumLike)
assert not speclike.background_spectrum.is_poisson
speclike = nai3.to_spectrumlike(extract_measured_background=True)
assert isinstance(speclike, DispersionSpectrumLike)
assert speclike.background_spectrum.is_poisson
# test binning
# should not have bins yet
with pytest.raises(RuntimeError):
nai3.bins
# First catch the errors
# This is without specifying the correct options name
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method='constant')
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method='significance')
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method='constant', p0=.1)
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method='significance', dt=1)
# now incorrect options
with pytest.raises(RuntimeError):
nai3.create_time_bins(start=0, stop=10, method='not_a_method')
# Now test values
nai3.create_time_bins(start=0, stop=10, method='constant', dt=1)
assert len(nai3.bins) == 10
assert nai3.bins.argsort() == range(len(nai3.bins))
nai3.create_time_bins(start=0, stop=10, method='bayesblocks', p0=.1)
assert nai3.bins.argsort() == range(len(nai3.bins))
assert len(nai3.bins) == 5
nai3.create_time_bins(start=0, stop=10, method='significance', sigma=40)
assert nai3.bins.argsort() == range(len(nai3.bins))
assert len(nai3.bins) == 5
nai3.view_lightcurve(use_binner=True)
nai3.write_pha_from_binner('test_from_nai3', overwrite=True)