def test_correlations_narrow_v1_vs_v5(self): inverse_power = wpe.get_power_inverse(self.Y) R_desired, r_desired = wpe.get_correlations_narrow( self.Y, inverse_power, self.K, self.delay) R_actual, r_actual = wpe.get_correlations_narrow_v5( 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(self): np_inv_power = wpe.get_power_inverse(self.Y) np_corr = wpe.get_correlations_narrow_v5(self.Y, np_inv_power, self.K, self.delay) with tf.Graph().as_default(), tf.Session() as sess: tf_signal = tf.placeholder(tf.complex128, shape=[None, None]) tf_inverse_power = tf_wpe.get_power_inverse(tf_signal) tf_res = tf_wpe.get_correlations_narrow(tf_signal, tf_inverse_power, self.K, self.delay) tf_corr = sess.run(tf_res, {tf_signal: self.Y}) np.testing.assert_allclose(np_corr[0], tf_corr[0]) np.testing.assert_allclose(np_corr[1], tf_corr[1])