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