9
9
import operator
10
10
from itertools import product , starmap
11
11
12
- from numpy import nan , inf
13
12
import numpy as np
14
13
import pandas as pd
15
14
@@ -670,7 +669,7 @@ def test_floordiv_div(self):
670
669
ser = Series ([- 1 , 0 , 1 ], name = 'first' )
671
670
672
671
result = ser // 0
673
- expected = Series ([- inf , nan , inf ], name = 'first' )
672
+ expected = Series ([- np . inf , np . nan , np . inf ], name = 'first' )
674
673
assert_series_equal (result , expected )
675
674
676
675
@@ -1069,17 +1068,17 @@ def test_timedelta64_ops_nat(self):
1069
1068
assert_series_equal (1.5 * timedelta_series ,
1070
1069
Series ([NaT , Timedelta ('1.5s' )]))
1071
1070
1072
- assert_series_equal (timedelta_series * nan ,
1071
+ assert_series_equal (timedelta_series * np . nan ,
1073
1072
nat_series_dtype_timedelta )
1074
- assert_series_equal (nan * timedelta_series ,
1073
+ assert_series_equal (np . nan * timedelta_series ,
1075
1074
nat_series_dtype_timedelta )
1076
1075
1077
1076
# division
1078
1077
assert_series_equal (timedelta_series / 2 ,
1079
1078
Series ([NaT , Timedelta ('0.5s' )]))
1080
1079
assert_series_equal (timedelta_series / 2.0 ,
1081
1080
Series ([NaT , Timedelta ('0.5s' )]))
1082
- assert_series_equal (timedelta_series / nan ,
1081
+ assert_series_equal (timedelta_series / np . nan ,
1083
1082
nat_series_dtype_timedelta )
1084
1083
1085
1084
def test_td64_sub_NaT (self ):
@@ -2183,12 +2182,12 @@ def _check_fill(meth, op, a, b, fill_value=0):
2183
2182
with np .errstate (all = 'ignore' ):
2184
2183
if amask [i ]:
2185
2184
if bmask [i ]:
2186
- exp_values .append (nan )
2185
+ exp_values .append (np . nan )
2187
2186
continue
2188
2187
exp_values .append (op (fill_value , b [i ]))
2189
2188
elif bmask [i ]:
2190
2189
if amask [i ]:
2191
- exp_values .append (nan )
2190
+ exp_values .append (np . nan )
2192
2191
continue
2193
2192
exp_values .append (op (a [i ], fill_value ))
2194
2193
else :
@@ -2198,8 +2197,8 @@ def _check_fill(meth, op, a, b, fill_value=0):
2198
2197
expected = Series (exp_values , exp_index )
2199
2198
assert_series_equal (result , expected )
2200
2199
2201
- a = Series ([nan , 1. , 2. , 3. , nan ], index = np .arange (5 ))
2202
- b = Series ([nan , 1 , nan , 3 , nan , 4. ], index = np .arange (6 ))
2200
+ a = Series ([np . nan , 1. , 2. , 3. , np . nan ], index = np .arange (5 ))
2201
+ b = Series ([np . nan , 1 , np . nan , 3 , np . nan , 4. ], index = np .arange (6 ))
2203
2202
2204
2203
pairings = []
2205
2204
for op in ['add' , 'sub' , 'mul' , 'pow' , 'truediv' , 'floordiv' ]:
0 commit comments