(method)
| 1374 | ], |
| 1375 | ) |
| 1376 | def test_quantile_missing(method): |
| 1377 | df = pd.DataFrame({"A": [0, np.nan, 2]}) |
| 1378 | # TODO: Test npartitions=2 |
| 1379 | # (see https://github.com/dask/dask/issues/9227) |
| 1380 | ddf = dd.from_pandas(df, npartitions=1) |
| 1381 | expected = df.quantile() |
| 1382 | result = ddf.quantile(method=method) |
| 1383 | assert_eq(result, expected) |
| 1384 | |
| 1385 | expected = df.A.quantile() |
| 1386 | result = ddf.A.quantile(method=method) |
| 1387 | assert_eq(result, expected) |
| 1388 | |
| 1389 | |
| 1390 | @pytest.mark.parametrize( |