Merge pull request #157 from fmieting/kernel_startup_timeout

Kernel startup timeout can be set on CLI

Author: vidartf

nbval/kernel.py

@@ -75,14 +75,15 @@ class RunningKernel(object):
     this class.
 
     """
-    def __init__(self, kernel_name, cwd=None):
+    def __init__(self, kernel_name, cwd=None, startup_timeout=60):
         """
         Initialise a new kernel
         specify that matplotlib is inline and connect the stderr.
         Stores the active kernel process and its manager.
         """
 
         self.km, self.kc = start_new_kernel(
+            startup_timeout=startup_timeout,
             kernel_name=kernel_name,
             stderr=open(os.devnull, 'w'),
             cwd=cwd,

nbval/plugin.py

@@ -93,6 +93,10 @@ def pytest_addoption(parser):
                     type=float,
                     help='Timeout for cell execution, in seconds.')
 
+    group.addoption('--nbval-kernel-startup-timeout', action='store', default=60,
+                    type=float,
+                    help='Timeout for kernel startup, in seconds.')
+
     term_group = parser.getgroup("terminal reporting")
     term_group._addoption(
         '--nbdime', action='store_true',
@@ -234,7 +238,11 @@ def setup(self):
         else:
             kernel_name = self.nb.metadata.get(
                 'kernelspec', {}).get('name', 'python')
-        self.kernel = RunningKernel(kernel_name, str(self.fspath.dirname))
+        self.kernel = RunningKernel(
+            kernel_name,
+            cwd=str(self.fspath.dirname),
+            startup_timeout=self.config.option.nbval_kernel_startup_timeout, 
+        )
         self.setup_sanitize_files()
         if getattr(self.parent.config.option, 'cov_source', None):
             setup_coverage(self.parent.config, self.kernel, getattr(self, "fspath", None))