def test_correlations_v1_vs_v2(self): inverse_power = wpe.get_power_inverse(self.Y) R_desired, r_desired = wpe.get_correlations(self.Y, inverse_power, self.K, self.delay) R_actual, r_actual = wpe.get_correlations_v2(self.Y, inverse_power, self.K, self.delay) tc.assert_allclose(R_actual, R_desired) tc.assert_allclose(r_actual, r_desired)
def test_correlations_v1_vs_v2_toy_example(self): K = 3 delay = 1 Y = np.asarray([[11, 12, 13, 14], [41, 22, 23, 24]], dtype=np.float32) inverse_power = wpe.get_power_inverse(Y) R_desired, r_desired = wpe.get_correlations(Y, inverse_power, K, delay) R_actual, r_actual = wpe.get_correlations_v2(Y, inverse_power, K, delay) tc.assert_allclose(R_actual, R_desired) tc.assert_allclose(r_actual, r_desired)