def test_task_args_sets_new_attrs_on_mapped_tasks(self, attr, val):
t = Task()
with Flow(name="test") as f:
t.map(upstream_tasks=[1, 2, 3, 4], task_args={attr: val})
tasks = f.get_tasks(name="Task")
assert all(getattr(tt, attr) == val for tt in tasks)
def test_task_check_mapped_args_are_subscriptable_in_advance(self):
t = Task()
with pytest.raises(TypeError):
with Flow(name="test"):
t.map({1, 2, 3, 4})
