Example #1
0
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')
Example #2
0
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')