@@ -380,8 +380,7 @@ def setup_method(self):
380380 @pytest .mark .parametrize ("norm" , [None , "forward" , "backward" , "ortho" ])
381381 @pytest .mark .parametrize ("order" , ["C" , "F" ])
382382 def test_fft2 (self , dtype , axes , norm , order ):
383- x = generate_random_numpy_array ((2 , 3 , 4 ), dtype )
384- a_np = numpy .array (x , order = order )
383+ a_np = generate_random_numpy_array ((2 , 3 , 4 ), dtype , order )
385384 a = dpnp .array (a_np )
386385
387386 result = dpnp .fft .fft2 (a , axes = axes , norm = norm )
@@ -443,8 +442,7 @@ def setup_method(self):
443442 @pytest .mark .parametrize ("norm" , [None , "backward" , "forward" , "ortho" ])
444443 @pytest .mark .parametrize ("order" , ["C" , "F" ])
445444 def test_fftn (self , dtype , axes , norm , order ):
446- x = generate_random_numpy_array ((2 , 3 , 4 , 5 ), dtype )
447- a_np = numpy .array (x , order = order )
445+ a_np = generate_random_numpy_array ((2 , 3 , 4 , 5 ), dtype , order )
448446 a = dpnp .array (a_np )
449447
450448 result = dpnp .fft .fftn (a , axes = axes , norm = norm )
@@ -698,8 +696,7 @@ def test_irfft_1D_on_2D_array(self, dtype, n, axis, norm, order):
698696 @pytest .mark .parametrize ("norm" , [None , "backward" , "forward" , "ortho" ])
699697 @pytest .mark .parametrize ("order" , ["C" , "F" ])
700698 def test_irfft_1D_on_3D_array (self , dtype , n , axis , norm , order ):
701- x = generate_random_numpy_array ((4 , 5 , 6 ), dtype )
702- a_np = numpy .array (x , order = order )
699+ a_np = generate_random_numpy_array ((4 , 5 , 6 ), dtype , order )
703700 # each 1-D array of input should be Hermitian
704701 if axis == 0 :
705702 a_np [0 ].imag = 0
@@ -936,8 +933,7 @@ def setup_method(self):
936933 @pytest .mark .parametrize ("norm" , [None , "backward" , "forward" , "ortho" ])
937934 @pytest .mark .parametrize ("order" , ["C" , "F" ])
938935 def test_rfft2 (self , dtype , axes , norm , order ):
939- x = generate_random_numpy_array ((2 , 3 , 4 ), dtype )
940- a_np = numpy .array (x , order = order )
936+ a_np = generate_random_numpy_array ((2 , 3 , 4 ), dtype , order )
941937 a = dpnp .asarray (a_np )
942938
943939 result = dpnp .fft .rfft2 (a , axes = axes , norm = norm )
@@ -1001,8 +997,7 @@ def setup_method(self):
1001997 @pytest .mark .parametrize ("norm" , [None , "backward" , "forward" , "ortho" ])
1002998 @pytest .mark .parametrize ("order" , ["C" , "F" ])
1003999 def test_rfftn (self , dtype , axes , norm , order ):
1004- x = generate_random_numpy_array ((2 , 3 , 4 , 5 ), dtype )
1005- a_np = numpy .array (x , order = order )
1000+ a_np = generate_random_numpy_array ((2 , 3 , 4 , 5 ), dtype , order )
10061001 a = dpnp .asarray (a_np )
10071002
10081003 result = dpnp .fft .rfftn (a , axes = axes , norm = norm )
0 commit comments