(self)
| 150 | lambda x: np.fft.rfft2(x, s=[x.shape[-2], x.shape[-1]]), _tf_fn) |
| 151 | |
| 152 | def testRFFT3D(self): |
| 153 | |
| 154 | def _to_expected(x): |
| 155 | return np.fft.rfftn( |
| 156 | x, axes=(-3, -2, -1), s=[x.shape[-3], x.shape[-2], x.shape[-1]]) |
| 157 | |
| 158 | def _tf_fn(x): |
| 159 | return signal.rfft3d( |
| 160 | x, fft_length=[x.shape[-3], x.shape[-2], x.shape[-1]]) |
| 161 | |
| 162 | self._VerifyFftMethod(INNER_DIMS_3D, np.real, _to_expected, _tf_fn) |
| 163 | |
| 164 | def testRFFT3DMismatchedSize(self): |
| 165 |
nothing calls this directly
no test coverage detected