(method)
| 1390 | ], |
| 1391 | ) |
| 1392 | def test_quantile_missing(method): |
| 1393 | df = pd.DataFrame({"A": [0, np.nan, 2]}) |
| 1394 | # TODO: Test npartitions=2 |
| 1395 | # (see https://github.com/dask/dask/issues/9227) |
| 1396 | ddf = dd.from_pandas(df, npartitions=1) |
| 1397 | expected = df.quantile() |
| 1398 | result = ddf.quantile(method=method) |
| 1399 | assert_eq(result, expected) |
| 1400 | |
| 1401 | expected = df.A.quantile() |
| 1402 | result = ddf.A.quantile(method=method) |
| 1403 | assert_eq(result, expected) |
| 1404 | |
| 1405 | |
| 1406 | @pytest.mark.parametrize( |