@@ -24,7 +24,7 @@ def setup_method(self, method):
24
24
np .random .seed (11235 )
25
25
nanops ._USE_BOTTLENECK = False
26
26
27
- arr_shape = (11 , 7 , 5 )
27
+ arr_shape = (11 , 7 )
28
28
29
29
self .arr_float = np .random .randn (* arr_shape )
30
30
self .arr_float1 = np .random .randn (* arr_shape )
@@ -68,21 +68,21 @@ def setup_method(self, method):
68
68
self .arr_nan_infj = self .arr_inf * 1j
69
69
self .arr_complex_nan_infj = np .vstack ([self .arr_complex , self .arr_nan_infj ])
70
70
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
73
73
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
78
78
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 ]
81
81
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 ]
86
86
87
87
def teardown_method (self , method ):
88
88
nanops ._USE_BOTTLENECK = use_bn
0 commit comments