Skip to content

Commit 0c5a533

Browse files
committed
added test
1 parent 3614159 commit 0c5a533

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

pandas/core/array_algos/quantile.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -184,5 +184,5 @@ def _quantile_ea_fallback(
184184
try:
185185
out = type(values)._from_sequence(res, dtype=values.dtype)
186186
except TypeError:
187-
out = np.atleast_2d(np.asarray(res))
187+
out = np.atleast_2d(np.asarray(res, dtype=np.float64))
188188
return out

pandas/tests/series/methods/test_quantile.py

+6
Original file line numberDiff line numberDiff line change
@@ -217,3 +217,9 @@ def test_quantile_empty(self):
217217
res = s.quantile([0.5])
218218
exp = Series([pd.NaT], index=[0.5])
219219
tm.assert_series_equal(res, exp)
220+
221+
@pytest.mark.parametrize("dtype", [int, float, "Int64"])
222+
def test_quantile_dtypes(self, dtype):
223+
result = Series([1, 2, 3], dtype=dtype).quantile(np.arange(0, 1, 0.25))
224+
expected = Series(np.arange(1, 3, 0.5), index=np.arange(0, 1, 0.25))
225+
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)