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)