Skip to content

Commit 7cf9a7a

Browse files
committed
remove non-standard imports of nan, inf
1 parent 619d2bb commit 7cf9a7a

File tree

1 file changed

+8
-9
lines changed

1 file changed

+8
-9
lines changed

pandas/tests/series/test_operators.py

+8-9
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
import operator
1010
from itertools import product, starmap
1111

12-
from numpy import nan, inf
1312
import numpy as np
1413
import pandas as pd
1514

@@ -670,7 +669,7 @@ def test_floordiv_div(self):
670669
ser = Series([-1, 0, 1], name='first')
671670

672671
result = ser // 0
673-
expected = Series([-inf, nan, inf], name='first')
672+
expected = Series([-np.inf, np.nan, np.inf], name='first')
674673
assert_series_equal(result, expected)
675674

676675

@@ -1069,17 +1068,17 @@ def test_timedelta64_ops_nat(self):
10691068
assert_series_equal(1.5 * timedelta_series,
10701069
Series([NaT, Timedelta('1.5s')]))
10711070

1072-
assert_series_equal(timedelta_series * nan,
1071+
assert_series_equal(timedelta_series * np.nan,
10731072
nat_series_dtype_timedelta)
1074-
assert_series_equal(nan * timedelta_series,
1073+
assert_series_equal(np.nan * timedelta_series,
10751074
nat_series_dtype_timedelta)
10761075

10771076
# division
10781077
assert_series_equal(timedelta_series / 2,
10791078
Series([NaT, Timedelta('0.5s')]))
10801079
assert_series_equal(timedelta_series / 2.0,
10811080
Series([NaT, Timedelta('0.5s')]))
1082-
assert_series_equal(timedelta_series / nan,
1081+
assert_series_equal(timedelta_series / np.nan,
10831082
nat_series_dtype_timedelta)
10841083

10851084
def test_td64_sub_NaT(self):
@@ -2183,12 +2182,12 @@ def _check_fill(meth, op, a, b, fill_value=0):
21832182
with np.errstate(all='ignore'):
21842183
if amask[i]:
21852184
if bmask[i]:
2186-
exp_values.append(nan)
2185+
exp_values.append(np.nan)
21872186
continue
21882187
exp_values.append(op(fill_value, b[i]))
21892188
elif bmask[i]:
21902189
if amask[i]:
2191-
exp_values.append(nan)
2190+
exp_values.append(np.nan)
21922191
continue
21932192
exp_values.append(op(a[i], fill_value))
21942193
else:
@@ -2198,8 +2197,8 @@ def _check_fill(meth, op, a, b, fill_value=0):
21982197
expected = Series(exp_values, exp_index)
21992198
assert_series_equal(result, expected)
22002199

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))
22032202

22042203
pairings = []
22052204
for op in ['add', 'sub', 'mul', 'pow', 'truediv', 'floordiv']:

0 commit comments

Comments
 (0)