def test_MLTask_no_combine_output(mocked_target, mltask_kwargs): mltask = MLTask(combine_outputs=False, **mltask_kwargs) mltask.output() args, kwargs = mocked_target._mock_call_args assert args[0].endswith('.length')
def test_MLTask_combine_output(mocked_target, mltask_kwargs): mltask = MLTask(combine_outputs=True, **mltask_kwargs) mltask.output() args, kwargs = mocked_target._mock_call_args assert args[0].endswith('.json')