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})