more

tests/test_optimize.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import awkward as ak
 import dask
 
 import dask_awkward as dak
@@ -11,5 +12,11 @@ def test_multiple_computes(pq_points_dir) -> None:
     # from_parquet so we get two unique collections.
     ds2 = dak.from_parquet(pq_points_dir, columns=["points"])
 
+    lists = [[[1, 2, 3], [4, 5]], [[], [0, 0, 0]]]
+    ds3 = dak.from_lists([[[1, 2, 3], [4, 5]], [[], [0, 0, 0]]])
+
     assert ds1.name != ds2.name
     assert dask.compute(ds1.points.x, ds2.points.y)
+    assert dask.compute(ds1.points)
+    things = dask.compute(ds1.points, ds2.points.x, ds2.points.y, ds1.points.y, ds3)
+    assert things[-1].tolist() == ak.Array(lists[0] + lists[1]).tolist()