예제 #1
0
 def test_rfftn_pair(self):
     test_axes = (0, -1)
     test_shape = (4, 16)
     my_rfftn, my_irfftn = rfftn(self.x3,
                                 shape=test_shape,
                                 axes=test_axes,
                                 fft_pair=True)
     x_fft = my_rfftn(self.x3)
     assert_array_almost_equal(my_irfftn(x_fft),
                               pad_array(self.x3, (4, 3, 16)))
예제 #2
0
 def test_rfftn_pair_cropped(self):
     test_axes = (0, -1)
     test_shape = (4, 16)
     my_rfftn, my_irfftn = rfftn(self.x3,
                                 shape=test_shape,
                                 axes=test_axes,
                                 fft_pair=True,
                                 crop_ifft=True)
     x_fft = my_rfftn(self.x3)
     assert_array_almost_equal(my_irfftn(x_fft), self.x3)
예제 #3
0
 def test_rfftn_3d_axes(self):
     test_axes = (0, -1)
     my_rfftn = rfftn(self.x3, axes=test_axes)
     assert_array_almost_equal(my_rfftn(self.x3),
                               np.fft.rfftn(self.x3, axes=test_axes))
예제 #4
0
 def test_rfftn_3d_mixed_and_short_shape(self):
     test_shape = (2, 16)
     my_rfftn = rfftn(self.x3, shape=test_shape)
     assert_array_almost_equal(my_rfftn(self.x3),
                               np.fft.rfftn(self.x3, s=test_shape))
예제 #5
0
 def test_rfftn_3d_smaller_shape(self):
     test_shape = (3, 3, 6)
     my_rfftn = rfftn(self.x3, shape=test_shape)
     assert_array_almost_equal(my_rfftn(self.x3),
                               np.fft.rfftn(self.x3, s=test_shape))
예제 #6
0
 def test_rfftn_3d_larger_shape(self):
     test_shape = (4, 6, 14)
     my_rfftn = rfftn(self.x3, shape=test_shape)
     assert_array_almost_equal(my_rfftn(self.x3),
                               np.fft.rfftn(self.x3, s=test_shape))
예제 #7
0
 def test_rfftn_2d(self):
     my_rfftn = rfftn(self.x2)
     assert_array_almost_equal(my_rfftn(self.x2), np.fft.rfftn(self.x2))
예제 #8
0
 def test_rfftn_3d_axes(self):
     test_axes = (0, -1)
     my_rfftn = rfftn(self.x3, axes=test_axes)
     assert_array_almost_equal(my_rfftn(self.x3),
                               np.fft.rfftn(self.x3, axes=test_axes))
예제 #9
0
 def test_rfftn_3d_mixed_and_short_shape(self):
     test_shape = (2, 16)
     my_rfftn = rfftn(self.x3, shape=test_shape)
     assert_array_almost_equal(my_rfftn(self.x3), np.fft.rfftn(self.x3,
                                                               s=test_shape))
예제 #10
0
 def test_rfftn_3d_smaller_shape(self):
     test_shape = (3, 3, 6)
     my_rfftn = rfftn(self.x3, shape=test_shape)
     assert_array_almost_equal(my_rfftn(self.x3), np.fft.rfftn(self.x3,
                                                               s=test_shape))
예제 #11
0
 def test_rfftn_3d_larger_shape(self):
     test_shape = (4, 6, 14)
     my_rfftn = rfftn(self.x3, shape=test_shape)
     assert_array_almost_equal(my_rfftn(self.x3), np.fft.rfftn(self.x3,
                                                               s=test_shape))
예제 #12
0
 def test_rfftn_2d(self):
     my_rfftn = rfftn(self.x2)
     assert_array_almost_equal(my_rfftn(self.x2), np.fft.rfftn(self.x2))