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