| 4576 | |
| 4577 | |
| 4578 | def test_mixed_dask_array_multi_dimensional(): |
| 4579 | df = pd.DataFrame( |
| 4580 | {"x": [1, 2, 3, 4, 5], "y": [5.0, 6.0, 7.0, 8.0, 9.0]}, columns=["x", "y"] |
| 4581 | ) |
| 4582 | ddf = dd.from_pandas(df, npartitions=2) |
| 4583 | |
| 4584 | x = (df.values + 1).astype(float) |
| 4585 | dx = (ddf.values + 1).astype(float) |
| 4586 | |
| 4587 | assert_eq(ddf + dx + 1, df + x + 1) |
| 4588 | assert_eq(ddf + dx.rechunk((None, 1)) + 1, df + x + 1) |
| 4589 | assert_eq(ddf[["y", "x"]] + dx + 1, df[["y", "x"]] + x + 1) |
| 4590 | |
| 4591 | |
| 4592 | @pytest.mark.skip_with_pyarrow_strings # DateOffset has to be an object |