def test_experiment_wrapper_method_call(self):
data = base64.b64encode(pickle.dumps(method_call)).decode('utf-8')
args = [
'',
'--args_data',
data,
'--log_dir',
'data/',
'--resume_from_dir',
'resume_dir/',
'--resume_from_epoch',
'first',
]
run_experiment(args)
def test_experiment_wrapper_invalid_method_call(self):
data = base64.b64encode(
pickle.dumps(invalid_method_call)).decode('utf-8')
args = ['', '--args_data', data]
with pytest.raises(BaseException):
run_experiment(args)