(df)
| 23 | |
| 24 | |
| 25 | def test_simple(df): |
| 26 | out = (df["x"] + df["y"]) - 1 |
| 27 | unfused = optimize(out, fuse=False) |
| 28 | fused = optimize(out, fuse=True) |
| 29 | |
| 30 | # Should only get one task per partition |
| 31 | # from_pandas is not fused together |
| 32 | assert len(fused.dask) == df.npartitions + 10 |
| 33 | assert_eq(fused, unfused) |
| 34 | |
| 35 | |
| 36 | def test_with_non_fusable_on_top(df): |