6
6
7
7
from pandas .core .dtypes .common import ensure_int64
8
8
9
- from pandas import Index , isna
9
+ from pandas import Index , Series , isna
10
10
from pandas .core .groupby .ops import generate_bins_generic
11
11
import pandas .util .testing as tm
12
12
from pandas .util .testing import assert_almost_equal
13
13
14
14
15
15
def test_series_grouper ():
16
- from pandas import Series
17
-
18
16
obj = Series (np .random .randn (10 ))
19
17
dummy = obj [:0 ]
20
18
@@ -31,8 +29,6 @@ def test_series_grouper():
31
29
32
30
33
31
def test_series_bin_grouper ():
34
- from pandas import Series
35
-
36
32
obj = Series (np .random .randn (10 ))
37
33
dummy = obj [:0 ]
38
34
@@ -123,30 +119,32 @@ class TestMoments:
123
119
124
120
class TestReducer :
125
121
def test_int_index (self ):
126
- from pandas .core .series import Series
127
-
128
122
arr = np .random .randn (100 , 4 )
129
- result = reduction .reduce (arr , np .sum , labels = Index (np .arange (4 )))
123
+ result = reduction .compute_reduction (arr , np .sum , labels = Index (np .arange (4 )))
130
124
expected = arr .sum (0 )
131
125
assert_almost_equal (result , expected )
132
126
133
- result = reduction .reduce (arr , np .sum , axis = 1 , labels = Index (np .arange (100 )))
127
+ result = reduction .compute_reduction (
128
+ arr , np .sum , axis = 1 , labels = Index (np .arange (100 ))
129
+ )
134
130
expected = arr .sum (1 )
135
131
assert_almost_equal (result , expected )
136
132
137
133
dummy = Series (0.0 , index = np .arange (100 ))
138
- result = reduction .reduce (arr , np .sum , dummy = dummy , labels = Index (np .arange (4 )))
134
+ result = reduction .compute_reduction (
135
+ arr , np .sum , dummy = dummy , labels = Index (np .arange (4 ))
136
+ )
139
137
expected = arr .sum (0 )
140
138
assert_almost_equal (result , expected )
141
139
142
140
dummy = Series (0.0 , index = np .arange (4 ))
143
- result = reduction .reduce (
141
+ result = reduction .compute_reduction (
144
142
arr , np .sum , axis = 1 , dummy = dummy , labels = Index (np .arange (100 ))
145
143
)
146
144
expected = arr .sum (1 )
147
145
assert_almost_equal (result , expected )
148
146
149
- result = reduction .reduce (
147
+ result = reduction .compute_reduction (
150
148
arr , np .sum , axis = 1 , dummy = dummy , labels = Index (np .arange (100 ))
151
149
)
152
150
assert_almost_equal (result , expected )
0 commit comments