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)
