def test_MLTask_requires_child(mocked_task, mltask_kwargs): mltask = MLTask(child={'job_name': 'something'}, **mltask_kwargs) task = mltask.requires() assert mocked_task.call_count == 1 assert task == mocked_task._mock_return_value
def test_MLTask_requires_no_child_yes_input(mltask_kwargs): mltask = MLTask(input_task=SomeTask, **mltask_kwargs) task = mltask.requires() assert type(task) == SomeTask