def test_fingerprint_relaxation(self): one_vec = np.ones(self.T.shape[0]) """k=None""" relax_amp = np.dot(self.p0, self.R) * np.dot(self.L, self.obs1) tsn, relax_ampn = fingerprint_relaxation(self.T, self.p0, self.obs1) assert_allclose(tsn, self.ts) assert_allclose(relax_ampn, relax_amp) """k=4""" k = self.k relax_amp = np.dot(self.p0, self.R[:, 0:k]) * np.dot(self.L[0:k, :], self.obs1) tsn, relax_ampn = fingerprint_relaxation(self.T, self.p0, self.obs1, k=k) assert_allclose(tsn, self.ts[0:k]) assert_allclose(relax_ampn, relax_amp)
def test_fingerprint_relaxation(self): one_vec = np.ones(self.T.shape[0]) relax_amp = np.dot(self.p0, self.R) * np.dot(self.L, self.obs1) tsn, relax_ampn = fingerprint_relaxation(self.T, self.p0, self.obs1, k=self.k) assert_allclose(tsn, self.ts) assert_allclose(relax_ampn, relax_amp)