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'