From bd025903a743f248206612f997e7a512e577f28f Mon Sep 17 00:00:00 2001 From: Simon Hawkins Date: Tue, 3 Dec 2019 13:55:19 +0000 Subject: [PATCH] TST: update test_nanops setup_method --- pandas/tests/test_nanops.py | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/pandas/tests/test_nanops.py b/pandas/tests/test_nanops.py index e5d963a307502..c207c803510ca 100644 --- a/pandas/tests/test_nanops.py +++ b/pandas/tests/test_nanops.py @@ -24,7 +24,7 @@ def setup_method(self, method): np.random.seed(11235) nanops._USE_BOTTLENECK = False - arr_shape = (11, 7, 5) + arr_shape = (11, 7) self.arr_float = np.random.randn(*arr_shape) self.arr_float1 = np.random.randn(*arr_shape) @@ -68,21 +68,21 @@ def setup_method(self, method): self.arr_nan_infj = self.arr_inf * 1j self.arr_complex_nan_infj = np.vstack([self.arr_complex, self.arr_nan_infj]) - self.arr_float_2d = self.arr_float[:, :, 0] - self.arr_float1_2d = self.arr_float1[:, :, 0] + self.arr_float_2d = self.arr_float + self.arr_float1_2d = self.arr_float1 - self.arr_nan_2d = self.arr_nan[:, :, 0] - self.arr_float_nan_2d = self.arr_float_nan[:, :, 0] - self.arr_float1_nan_2d = self.arr_float1_nan[:, :, 0] - self.arr_nan_float1_2d = self.arr_nan_float1[:, :, 0] + self.arr_nan_2d = self.arr_nan + self.arr_float_nan_2d = self.arr_float_nan + self.arr_float1_nan_2d = self.arr_float1_nan + self.arr_nan_float1_2d = self.arr_nan_float1 - self.arr_float_1d = self.arr_float[:, 0, 0] - self.arr_float1_1d = self.arr_float1[:, 0, 0] + self.arr_float_1d = self.arr_float[:, 0] + self.arr_float1_1d = self.arr_float1[:, 0] - self.arr_nan_1d = self.arr_nan[:, 0, 0] - self.arr_float_nan_1d = self.arr_float_nan[:, 0, 0] - self.arr_float1_nan_1d = self.arr_float1_nan[:, 0, 0] - self.arr_nan_float1_1d = self.arr_nan_float1[:, 0, 0] + self.arr_nan_1d = self.arr_nan[:, 0] + self.arr_float_nan_1d = self.arr_float_nan[:, 0] + self.arr_float1_nan_1d = self.arr_float1_nan[:, 0] + self.arr_nan_float1_1d = self.arr_nan_float1[:, 0] def teardown_method(self, method): nanops._USE_BOTTLENECK = use_bn