def fit_eval_periodogram(method):
my_per = periodogram(method=method)
mjd, mag, err = create_test_data()
my_per.set_data(mjd, mag, err)
my_per.frequency_grid_evaluation(fmin=0.01, fmax=10., fresolution=1e-4)
my_per.finetune_best_frequencies(n_local_optima=3, fresolution=1e-5)
return my_per.get_best_frequencies()
def test_standardize(self):
my_per = periodogram(method='AOV')
my_per.set_data(self.mjd, self.mag, self.err, standardize=True)
assert_allclose(my_per.mag[:3],
np.array([-0.885704, 0.472172, 0.9917],
dtype=np.float32),
atol=1e-4)
assert_allclose(my_per.err[:3],
np.array([1.439618, 0.724181, 1.507756],
dtype=np.float32),
atol=1e-4)
def test_standardize(tol=1e-7):
my_per = periodogram(method='AOV')
mjd, mag, err = create_test_data()
my_per.set_data(mjd, mag, err, standardize=True)
assert_allclose(my_per.mag[:3],
np.array([-0.87692106, 0.44308114, 0.94811785],
dtype=np.float32),
atol=tol)
assert_allclose(my_per.err[:3],
np.array([1.399464, 0.70398176, 1.4657015],
dtype=np.float32),
atol=tol)
def test_removenan():
my_per = periodogram(method='AOV')
mjd, mag, err = create_test_data()
mjd[1], mag[5], err[8] = np.nan, np.nan, np.nan
my_per.set_data(mjd, mag, err, remove_nan=True)
assert len(my_per.mag) == len(mjd) - 3
def test_removenan(self):
my_per = periodogram(method='AOV')
self.mjd[1], self.mag[5], self.err[8] = np.nan, np.nan, np.nan
my_per.set_data(self.mjd, self.mag, self.err, remove_nan=True)
self.assertEqual(len(my_per.mag), len(self.mjd) - 3)