Skip to content

Commit f7dea34

Browse files
simonjayhawkinsjreback
authored andcommitted
TST: update test_nanops setup_method (#29998)
1 parent 9d35623 commit f7dea34

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

pandas/tests/test_nanops.py

+13-13
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)