From 1a4d9873629bb28e19a124e40b063879fc5e3ad0 Mon Sep 17 00:00:00 2001 From: Tom Augspurger Date: Wed, 25 Apr 2018 15:50:48 -0500 Subject: [PATCH] Pass an array --- pandas/tests/test_algos.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pandas/tests/test_algos.py b/pandas/tests/test_algos.py index f99f18ac4b9bb..3c8c91ab30d3f 100644 --- a/pandas/tests/test_algos.py +++ b/pandas/tests/test_algos.py @@ -1597,6 +1597,7 @@ def test_take_empty(self, allow_fill): algos.take(arr, [0], allow_fill=allow_fill) def test_take_na_empty(self): - result = algos.take([], [-1, -1], allow_fill=True, fill_value=0) - expected = np.array([0, 0], dtype=np.int64) + result = algos.take(np.array([]), [-1, -1], allow_fill=True, + fill_value=0.0) + expected = np.array([0., 0.]) tm.assert_numpy_array_equal(result, expected)