(
npartitions, enforce_metadata, transform_divisions, align_dataframes
)
| 158 | @pytest.mark.parametrize("transform_divisions", [True, False]) |
| 159 | @pytest.mark.parametrize("align_dataframes", [True, False]) |
| 160 | def test_map_overlap_names( |
| 161 | npartitions, enforce_metadata, transform_divisions, align_dataframes |
| 162 | ): |
| 163 | ddf = dd.from_pandas(df, npartitions) |
| 164 | |
| 165 | res = ddf.map_overlap( |
| 166 | shifted_sum, |
| 167 | 0, |
| 168 | 3, |
| 169 | 0, |
| 170 | 3, |
| 171 | c=2, |
| 172 | align_dataframes=align_dataframes, |
| 173 | transform_divisions=transform_divisions, |
| 174 | enforce_metadata=enforce_metadata, |
| 175 | ) |
| 176 | res2 = ddf.map_overlap( |
| 177 | shifted_sum, |
| 178 | 0, |
| 179 | 3, |
| 180 | 0, |
| 181 | 3, |
| 182 | c=2, |
| 183 | align_dataframes=align_dataframes, |
| 184 | transform_divisions=transform_divisions, |
| 185 | enforce_metadata=enforce_metadata, |
| 186 | ) |
| 187 | assert set(res.dask) == set(res2.dask) |
| 188 | |
| 189 | res3 = ddf.map_overlap( |
| 190 | shifted_sum, |
| 191 | 0, |
| 192 | 3, |
| 193 | 0, |
| 194 | 3, |
| 195 | c=3, |
| 196 | align_dataframes=align_dataframes, |
| 197 | transform_divisions=transform_divisions, |
| 198 | enforce_metadata=enforce_metadata, |
| 199 | ) |
| 200 | assert res3._name != res._name |
| 201 | # Difference is just the final map |
| 202 | diff = res3.dask.keys() - res.dask.keys() |
| 203 | assert len(diff) == npartitions |
| 204 | |
| 205 | res4 = ddf.map_overlap(shifted_sum, 3, 0, 0, 3, c=2) |
| 206 | assert res4._name != res._name |
| 207 | |
| 208 | |
| 209 | def test_map_overlap_errors(): |
nothing calls this directly
no test coverage detected