def test_z_n_search_expocorr_fails(self): """Test pulse phase calculation, frequency only.""" frequencies = np.arange(9.8, 9.99, 0.1/self.tseg) with pytest.raises(ValueError) as excinfo: freq, stat = z_n_search(self.event_times, frequencies, nharm=1, nbin=35, expocorr=True) assert 'To calculate exposure correction' in str(excinfo.value)
def test_z_n_search_weights(self): """Test pulse phase calculation, frequency only.""" frequencies = np.arange(9.8, 9.99, 0.1/self.tseg) freq, stat = z_n_search(self.times, frequencies, nbin=44, nharm=1, weights=self.counts) minbin = np.argmin(np.abs(frequencies - self.pulse_frequency)) maxstatbin = freq[np.argmax(stat)] assert np.allclose(maxstatbin, frequencies[minbin], atol=0.1/self.tseg)
def test_z_n_search_expocorr(self): """Test pulse phase calculation, frequency only.""" frequencies = np.arange(9.8, 9.99, 0.1/self.tseg) with pytest.warns(UserWarning): freq, stat = z_n_search(self.event_times, frequencies, nbin=64, nharm=2, expocorr=True, gti=self.gti) minbin = np.argmin(np.abs(frequencies - self.pulse_frequency)) maxstatbin = freq[np.argmax(stat)] assert np.allclose(maxstatbin, frequencies[minbin], atol=0.1/self.tseg)
def test_z_n_search_fdot(self): """Test pulse phase calculation, frequency only.""" frequencies = np.arange(9.8, 9.99, 0.1/self.tseg) fdots = [-0.1, 0, 0.1] freq, fdot, stat = z_n_search(self.event_times, frequencies, nbin=25, nharm=2, fdots=fdots) minbin = np.argmin(np.abs(frequencies - self.pulse_frequency)) maxstatbin = freq.flatten()[np.argmax(stat)] assert np.allclose(maxstatbin, frequencies[minbin], atol=0.1/self.tseg) maxfdot = fdot.flatten()[np.argmax(stat)] assert np.allclose(maxfdot, 0.0, atol=0.1/self.tseg)