def test_auto_correlation(self): result = np.array([1.68, -3.36, 5.2, -3.36, 1.68]) lags_result = np.array([-2, -1, 0, 1, 2]) ac = AutoCorrelation(self.lc1) assert np.allclose(ac.lc1, self.lc1) assert np.allclose(ac.lc2, self.lc1) assert np.allclose(ac.corr, result) assert np.isclose(ac.dt, self.lc1.dt) assert ac.n == 5 assert np.allclose(ac.time_lags, lags_result) assert np.isclose(ac.time_shift, 0.0) assert ac.mode == 'same' assert ac.auto is True
def test_auto_correlation_with_full_mode(self): result = np.array([0.56, -1.48, 1.68, -3.36, 5.2, -3.36, 1.68, -1.48, 0.56]) lags_result = np.array([-4, -3, -2, -1, 0, 1, 2, 3, 4]) ac = AutoCorrelation(self.lc1, mode='full') assert np.allclose(ac.lc1, self.lc1) assert np.allclose(ac.lc2, self.lc1) assert np.allclose(ac.corr, result) assert np.isclose(ac.dt, self.lc1.dt) assert ac.n == 9 assert np.allclose(ac.time_lags, lags_result) assert np.isclose(ac.time_shift, 0.0) assert ac.mode == 'full' assert ac.auto is True