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)