def test_cross_sum_fft_n(self): out = core.cross_sum_fft(a,t1,t2, n = 3) self.assertTrue(np.allclose(out,cross_sum_a_t1_t2[0:3])) out = core.cross_sum_fft(a,t1,t2, aout = out) self.assertTrue(np.allclose(out,cross_sum_a_t1_t2[0:3]*2)) out = core.cross_sum_fft(a,t1,t2, n =3, aout = out) self.assertTrue(np.allclose(out,cross_sum_a_t1_t2[0:3]*3))
def test_cross_sum_equivalence_ND(self): for axis in (0,1,2): t1 = np.arange(test_data1.shape[axis]) t2 = np.arange(test_data1.shape[axis]) + 3 out1 = core.cross_sum(test_data1,t1,t2, axis = axis) out2 = core.cross_sum_fft(test_data1,t1,t2, axis = axis) self.assertTrue(np.allclose(out1,out2))
def test_cross_sum_fft(self): out = core.cross_sum_fft(a,t1,t2) self.assertTrue(np.allclose(out,cross_sum_a_t1_t2)) out = core.cross_sum_fft(a,t1,t2, aout = out) self.assertTrue(np.allclose(out,cross_sum_a_t1_t2*2))