Skip to content

Commit 7516ec7

Browse files
committed
Merge pull request pandas-dev#10270 from jreback/fix
TST: fix for bottleneck >= 1.0 nansum behavior, xref pandas-dev#9422
2 parents efc4a08 + e9f83ce commit 7516ec7

File tree

2 files changed

+17
-7
lines changed

2 files changed

+17
-7
lines changed

pandas/core/nanops.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
ensure_float, _ensure_float64,
1717
_ensure_int64, _ensure_object,
1818
is_float, is_integer, is_complex,
19-
is_float_dtype, is_floating_dtype,
19+
is_float_dtype,
2020
is_complex_dtype, is_integer_dtype,
2121
is_bool_dtype, is_object_dtype,
2222
is_datetime64_dtype, is_timedelta64_dtype,
@@ -373,7 +373,7 @@ def nansem(values, axis=None, skipna=True, ddof=1):
373373
var = nanvar(values, axis, skipna, ddof=ddof)
374374

375375
mask = isnull(values)
376-
if not is_floating_dtype(values):
376+
if not is_float_dtype(values.dtype):
377377
values = values.astype('f8')
378378
count, _ = _get_counts_nanvar(mask, axis, ddof)
379379

@@ -467,7 +467,7 @@ def nanargmin(values, axis=None, skipna=True):
467467
def nanskew(values, axis=None, skipna=True):
468468

469469
mask = isnull(values)
470-
if not is_floating_dtype(values):
470+
if not is_float_dtype(values.dtype):
471471
values = values.astype('f8')
472472

473473
count = _get_counts(mask, axis)
@@ -502,7 +502,7 @@ def nanskew(values, axis=None, skipna=True):
502502
def nankurt(values, axis=None, skipna=True):
503503

504504
mask = isnull(values)
505-
if not is_floating_dtype(values):
505+
if not is_float_dtype(values.dtype):
506506
values = values.astype('f8')
507507

508508
count = _get_counts(mask, axis)

pandas/tests/test_series.py

+13-3
Original file line numberDiff line numberDiff line change
@@ -2316,7 +2316,7 @@ def test_iteritems(self):
23162316
self.assertFalse(hasattr(self.series.iteritems(), 'reverse'))
23172317

23182318
def test_sum(self):
2319-
self._check_stat_op('sum', np.sum)
2319+
self._check_stat_op('sum', np.sum, check_allna=True)
23202320

23212321
def test_sum_inf(self):
23222322
import pandas.core.nanops as nanops
@@ -2629,7 +2629,7 @@ def test_npdiff(self):
26292629
r = np.diff(s)
26302630
assert_series_equal(Series([nan, 0, 0, 0, nan]), r)
26312631

2632-
def _check_stat_op(self, name, alternate, check_objects=False):
2632+
def _check_stat_op(self, name, alternate, check_objects=False, check_allna=False):
26332633
import pandas.core.nanops as nanops
26342634

26352635
def testit():
@@ -2653,7 +2653,17 @@ def testit():
26532653
assert_almost_equal(f(self.series), alternate(nona.values))
26542654

26552655
allna = self.series * nan
2656-
self.assertTrue(np.isnan(f(allna)))
2656+
2657+
if check_allna:
2658+
# xref 9422
2659+
# bottleneck >= 1.0 give 0.0 for an allna Series sum
2660+
try:
2661+
self.assertTrue(nanops._USE_BOTTLENECK)
2662+
import bottleneck as bn
2663+
self.assertTrue(bn.__version__ >= LooseVersion('1.0'))
2664+
self.assertEqual(f(allna),0.0)
2665+
except:
2666+
self.assertTrue(np.isnan(f(allna)))
26572667

26582668
# dtype=object with None, it works!
26592669
s = Series([1, 2, 3, None, 5])

0 commit comments

Comments
 (0)