Example #1
0
 def test_ifftn_pair_cropped(self):
     my_ifftn, my_fftn = ifftn(self.x3_fft,
                               shape=(8, 4, 32),
                               fft_pair=True,
                               crop_fft=True)
     my_x = my_ifftn(self.x3_fft)
     assert_array_almost_equal(my_fftn(my_x),
                               self.x3_fft)
Example #2
0
 def test_ifftn_3d_axes(self):
     test_axes = (0, -1)
     my_ifftn = ifftn(self.x3_fft, axes=test_axes)
     assert_array_almost_equal(my_ifftn(self.x3_fft),
                               np.fft.ifftn(self.x3_fft, axes=test_axes))
Example #3
0
 def test_ifftn_3d_mixed_and_short_shape(self):
     test_shape = (2, 16)
     my_ifftn = ifftn(self.x3_fft, shape=test_shape)
     assert_array_almost_equal(my_ifftn(self.x3_fft),
                               np.fft.ifftn(self.x3_fft, s=test_shape))
Example #4
0
 def test_ifftn_3d_smaller_shape(self):
     test_shape = (3, 3, 6)
     my_ifftn = ifftn(self.x3_fft, shape=test_shape)
     assert_array_almost_equal(my_ifftn(self.x3_fft),
                               np.fft.ifftn(self.x3_fft, s=test_shape))
Example #5
0
 def test_ifftn_3d_larger_shape(self):
     test_shape = (4, 6, 14)
     my_ifftn = ifftn(self.x3_fft, shape=test_shape)
     assert_array_almost_equal(my_ifftn(self.x3_fft),
                               np.fft.ifftn(self.x3_fft, s=test_shape))
Example #6
0
 def test_ifftn_2d(self):
     my_ifftn = ifftn(self.x2_fft)
     assert_array_almost_equal(my_ifftn(self.x2_fft),
                               np.fft.ifftn(self.x2_fft))
Example #7
0
 def test_ifftn_pair(self):
     my_ifftn, my_fftn = ifftn(self.x3_fft,
                               fft_pair=True)
     my_x = my_ifftn(self.x3_fft)
     assert_array_almost_equal(my_fftn(my_x),
                               self.x3_fft)
Example #8
0
 def test_ifftn_3d_axes(self):
     test_axes = (0, -1)
     my_ifftn = ifftn(self.x3_fft, axes=test_axes)
     assert_array_almost_equal(my_ifftn(self.x3_fft),
                               np.fft.ifftn(self.x3_fft, axes=test_axes))
Example #9
0
 def test_ifftn_3d_mixed_and_short_shape(self):
     test_shape = (2, 16)
     my_ifftn = ifftn(self.x3_fft, shape=test_shape)
     assert_array_almost_equal(my_ifftn(self.x3_fft),
                               np.fft.ifftn(self.x3_fft, s=test_shape))
Example #10
0
 def test_ifftn_3d_smaller_shape(self):
     test_shape = (3, 3, 6)
     my_ifftn = ifftn(self.x3_fft, shape=test_shape)
     assert_array_almost_equal(my_ifftn(self.x3_fft),
                               np.fft.ifftn(self.x3_fft, s=test_shape))
Example #11
0
 def test_ifftn_3d_larger_shape(self):
     test_shape = (4, 6, 14)
     my_ifftn = ifftn(self.x3_fft, shape=test_shape)
     assert_array_almost_equal(my_ifftn(self.x3_fft),
                               np.fft.ifftn(self.x3_fft, s=test_shape))
Example #12
0
 def test_ifftn_2d(self):
     my_ifftn = ifftn(self.x2_fft)
     assert_array_almost_equal(my_ifftn(self.x2_fft),
                               np.fft.ifftn(self.x2_fft))