def test_phaseogram_mjdref_pepoch(self):
phaseogr, phases, times, additional_info = \
phaseogram(self.event_times, self.pulse_frequency,
mjdref=57000, out_filename='phaseogram_mjdref.png',
pepoch=57000)
assert np.all(times >= 57000)
assert np.all((phases >= 0) & (phases <= 2))
def test_plot_phaseogram_direct(self):
import matplotlib.pyplot as plt
phaseogr, phases, times, additional_info = \
phaseogram(self.event_times, self.pulse_frequency)
plot_phaseogram(phaseogr, phases, times)
plt.savefig('phaseogram_direct.png')
plt.close(plt.gcf())
def test_phaseogram(self):
phaseogr, phases, times, additional_info = \
phaseogram(self.event_times, self.pulse_frequency)
assert np.all(times < 25.6)
assert np.any(times > 25)
assert np.all((phases >= 0) & (phases <= 2))
plt.close(plt.gcf())
def test_phaseogram_mjdref(self):
import matplotlib.pyplot as plt
phaseogr, phases, times, additional_info = \
phaseogram(self.event_times, self.pulse_frequency,
mjdref=57000, out_filename='phaseogram_mjdref.png')
assert np.all(times >= 57000)
assert np.all((phases >= 0) & (phases <= 2))
plt.close(plt.gcf())
def test_plot_phaseogram_fromfunc(self):
fig = plt.figure('Phaseogram from func')
ax = plt.subplot()
phaseogr, phases, times, additional_info = \
phaseogram(self.event_times, self.pulse_frequency, mjdref=57000,
pepoch=57000, phaseogram_ax=ax, plot=True)
plt.savefig('phaseogram_fromfunc.png')
plt.close(fig)
def test_phaseogram_weights(self):
phaseogr, phases, times, additional_info = \
phaseogram(self.times, self.pulse_frequency, weights=self.counts,
nph=16)
assert np.all(times < 25.6)
assert np.any(times > 25)
assert np.all((phases >= 0) & (phases <= 2))
fig = plt.figure('Phaseogram direct weights')
plot_phaseogram(phaseogr, phases, times)
plt.savefig('phaseogram_weights.png')
plt.close(fig)
def test_phaseogram_bad_weights(self):
with pytest.raises(ValueError) as excinfo:
phaseogr, phases, times, additional_info = \
phaseogram(self.event_times, self.pulse_frequency,
weights=[0, 2])
assert 'must match' in str(excinfo.value)
