Custom script

dask_drmaa/core.py

@@ -99,9 +99,10 @@ def remove_script():
                 if os.path.exists(fn):
                     os.remove(fn)
 
-            os.chmod(self.script, 0o777)
+        else:
+            self.script = os.path.abspath(script)
 
-        # TODO: check that user-provided script is executable
+        os.chmod(self.script, 0o777)
 
         self.template = merge(default_template,
                               {'remoteCommand': self.script},

dask_drmaa/tests/test_core.py

@@ -3,6 +3,7 @@
 
 import pytest
 
+from dask.utils import filetext
 from dask_drmaa import DRMAACluster
 from distributed import Client
 from distributed.utils_test import loop, inc
@@ -121,3 +122,22 @@ def cleanup_logs():
 
     import atexit
     atexit.register(cleanup_logs)
+
+
+def test_script(loop):
+    from dask_drmaa.core import script_template
+
+    script_template2 = script_template.replace('--name ', '--name my-worker-')
+
+    with filetext(script_template2, extension='sh', mode='wt') as fn:
+        with DRMAACluster(scheduler_port=0, script=fn) as cluster:
+            assert cluster.script == fn
+            cluster.start_workers(2)
+
+            start = time()
+            while len(cluster.scheduler.workers) < 2:
+                sleep(0.1)
+                assert time() < start + 10
+
+            assert all(info['name'].startswith('my-worker')
+                       for info in cluster.scheduler.worker_info.values())