(npartitions, use_dask_input)
| 57 | @pytest.mark.parametrize("npartitions", [1, 4]) |
| 58 | @pytest.mark.parametrize("use_dask_input", [True, False]) |
| 59 | def test_map_overlap(npartitions, use_dask_input): |
| 60 | ddf = df |
| 61 | if use_dask_input: |
| 62 | ddf = dd.from_pandas(df, npartitions) |
| 63 | |
| 64 | for before, after in [(0, 3), (3, 0), (3, 3), (0, 0)]: |
| 65 | # DataFrame |
| 66 | res = dd.map_overlap(shifted_sum, ddf, before, after, before, after, c=2) |
| 67 | sol = shifted_sum(df, before, after, c=2) |
| 68 | assert_eq(res, sol) |
| 69 | |
| 70 | # Series |
| 71 | res = dd.map_overlap(shifted_sum, ddf.b, before, after, before, after, c=2) |
| 72 | sol = shifted_sum(df.b, before, after, c=2) |
| 73 | assert_eq(res, sol) |
| 74 | |
| 75 | |
| 76 | @pytest.mark.parametrize("use_dask_input", [True, False]) |
nothing calls this directly
no test coverage detected