Exemplo n.º 1
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)
Exemplo n.º 2
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)))
Exemplo n.º 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))
Exemplo n.º 4
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))
Exemplo n.º 5
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))
Exemplo n.º 6
0
 def test_rfftn_2d(self):
     my_rfftn = rfftn(self.x2)
     assert_array_almost_equal(my_rfftn(self.x2), np.fft.rfftn(self.x2))
Exemplo n.º 7
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))
Exemplo n.º 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))
Exemplo n.º 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))
Exemplo n.º 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))
Exemplo n.º 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))
Exemplo n.º 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))