def testLocalRunScriptWithExec(self):
s = BytesIO(script2)
self.assertEqual(
run_script(s,
n_workers=2,
run_kwargs={'n_parallel': 2},
mode='exec')['status'], 'ok')
def test_local_run_script_with_data(setup_cluster):
s = BytesIO(script3)
data = {
'tensor': mt.arange(10),
'df': md.DataFrame({'s': mt.arange(9, 0, -1)})
}
assert run_script(
s,
data=data,
n_workers=1,
).fetch()['status'] == 'ok'
pytest.raises(TypeError, run_script, s, data=[])
def test_run_with_file(setup):
path = os.path.join(os.path.dirname(os.path.abspath(__file__)),
'sample_script.py')
assert run_script(path, n_workers=2)['status'] == 'ok'
def test_local_run_script_with_exec(setup):
s = BytesIO(script2)
assert run_script(s, n_workers=2, mode='exec')['status'] == 'ok'
def test_local_run_script(setup):
s = BytesIO(script1)
assert run_script(s, n_workers=2)['status'] == 'ok'
def testLocalRunScript(self):
s = BytesIO(script1)
self.assertEqual(
run_script(s, n_workers=2, run_kwargs={'n_parallel': 2})['status'],
'ok')
def test_local_run_script_with_exec(setup_cluster):
s = BytesIO(script2)
assert run_script(
s,
n_workers=2,
).fetch()['status'] == 'ok'
def test_local_run_script(setup_cluster):
s = BytesIO(script1)
assert run_script(s, n_workers=2).fetch()['status'] == 'ok'