def testToRealMatlab(self): dfts = [] dfts.append(([[4, 0, 1, 0]], [[2, 1, 0, 0]])) dfts.append(([[1, 0, 0, 1]], [[0, 1, 1, 0]])) dft_arr = mat.to_real_matlab_format(dfts) chan1 = np.array([4, 1j, 1, 0, 2], dtype=np.complex64) chan2 = np.array([1, 1j, 1j, 1, 0], dtype=np.complex64) print dft_arr self.assertListFloatEqual(chan1, dft_arr[0, :]) self.assertListFloatEqual(chan2, dft_arr[1, :])
def get_3d_real_distribution(self, dfts): rffts = mat.to_real_matlab_format(dfts) return self.get_distribution_real(rffts)