def test_rebin_small_df(self): segment_size = 3 dps = DynamicalPowerspectrum(self.lc, segment_size=segment_size) with pytest.raises(ValueError): dps.rebin_frequency(df_new=dps.df / 2.0)
def test_rebin_small_dt(self): segment_size = 3 dps = DynamicalPowerspectrum(self.lc_test, segment_size=segment_size) with pytest.raises(ValueError): dps.rebin_time(dt_new=2.0)
def test_size_of_trace_maximum(self): dps = DynamicalPowerspectrum(self.lc, segment_size=3) max_pos = dps.trace_maximum() nsegs = int(self.lc.tseg / dps.segment_size) assert len(max_pos) == nsegs
def test_trace_maximum_without_boundaries(self): dps = DynamicalPowerspectrum(self.lc, segment_size=3) max_pos = dps.trace_maximum() assert max(dps.freq[max_pos]) <= 1 / self.lc.dt assert min(dps.freq[max_pos]) >= 1 / dps.segment_size
def test_matrix(self): dps = DynamicalPowerspectrum(self.lc, segment_size=3) nsegs = int(self.lc.tseg / dps.segment_size) nfreq = int((1 / self.lc.dt) / (2 * (dps.freq[1] - dps.freq[0])) - (1 / self.lc.tseg)) assert dps.dyn_ps.shape == (nfreq, nsegs)
def test_with_long_seg_size(self): with pytest.raises(ValueError): dps = DynamicalPowerspectrum(self.lc, segment_size=1000)