Skip to content

Commit 441d251

Browse files
committed
Fixed test to use quantile parameter
1 parent 446ec4e commit 441d251

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pandas/tests/groupby/test_quantile.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,6 +237,6 @@ def test_groupby_quantile_nullable_array(values, q):
237237
@pytest.mark.parametrize("q", [0.5, [0.0, 0.5, 1.0]])
238238
def test_groupby_quantile_skips_invalid_dtype(q):
239239
df = pd.DataFrame({"a": [1], "b": [2.0], "c": ["x"]})
240-
result = df.groupby("a").quantile(0.5)
241-
expected = df.set_index("a")[["b"]]
240+
result = df.groupby("a").quantile(q)
241+
expected = df.groupby("a")[["b"]].quantile(q)
242242
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)