()
| 1230 | |
| 1231 | |
| 1232 | def test_from_delayed_iterator(): |
| 1233 | from dask.delayed import delayed |
| 1234 | |
| 1235 | def lazy_records(n): |
| 1236 | return ({"operations": [1, 2]} for _ in range(n)) |
| 1237 | |
| 1238 | delayed_records = delayed(lazy_records, pure=False) |
| 1239 | bag = db.from_delayed([delayed_records(5) for _ in range(5)]) |
| 1240 | assert db.compute( |
| 1241 | bag.count(), |
| 1242 | bag.pluck("operations").count(), |
| 1243 | bag.pluck("operations").flatten().count(), |
| 1244 | scheduler="sync", |
| 1245 | ) == (25, 25, 50) |
| 1246 | |
| 1247 | |
| 1248 | def test_range(): |