@@ -24,7 +24,7 @@ def setup_method(self, method):
2424 np .random .seed (11235 )
2525 nanops ._USE_BOTTLENECK = False
2626
27- arr_shape = (11 , 7 , 5 )
27+ arr_shape = (11 , 7 )
2828
2929 self .arr_float = np .random .randn (* arr_shape )
3030 self .arr_float1 = np .random .randn (* arr_shape )
@@ -68,21 +68,21 @@ def setup_method(self, method):
6868 self .arr_nan_infj = self .arr_inf * 1j
6969 self .arr_complex_nan_infj = np .vstack ([self .arr_complex , self .arr_nan_infj ])
7070
71- self .arr_float_2d = self .arr_float [:, :, 0 ]
72- self .arr_float1_2d = self .arr_float1 [:, :, 0 ]
71+ self .arr_float_2d = self .arr_float
72+ self .arr_float1_2d = self .arr_float1
7373
74- self .arr_nan_2d = self .arr_nan [:, :, 0 ]
75- self .arr_float_nan_2d = self .arr_float_nan [:, :, 0 ]
76- self .arr_float1_nan_2d = self .arr_float1_nan [:, :, 0 ]
77- self .arr_nan_float1_2d = self .arr_nan_float1 [:, :, 0 ]
74+ self .arr_nan_2d = self .arr_nan
75+ self .arr_float_nan_2d = self .arr_float_nan
76+ self .arr_float1_nan_2d = self .arr_float1_nan
77+ self .arr_nan_float1_2d = self .arr_nan_float1
7878
79- self .arr_float_1d = self .arr_float [:, 0 , 0 ]
80- self .arr_float1_1d = self .arr_float1 [:, 0 , 0 ]
79+ self .arr_float_1d = self .arr_float [:, 0 ]
80+ self .arr_float1_1d = self .arr_float1 [:, 0 ]
8181
82- self .arr_nan_1d = self .arr_nan [:, 0 , 0 ]
83- self .arr_float_nan_1d = self .arr_float_nan [:, 0 , 0 ]
84- self .arr_float1_nan_1d = self .arr_float1_nan [:, 0 , 0 ]
85- self .arr_nan_float1_1d = self .arr_nan_float1 [:, 0 , 0 ]
82+ self .arr_nan_1d = self .arr_nan [:, 0 ]
83+ self .arr_float_nan_1d = self .arr_float_nan [:, 0 ]
84+ self .arr_float1_nan_1d = self .arr_float1_nan [:, 0 ]
85+ self .arr_nan_float1_1d = self .arr_nan_float1 [:, 0 ]
8686
8787 def teardown_method (self , method ):
8888 nanops ._USE_BOTTLENECK = use_bn
0 commit comments