From 7927122d11082c98962b87714e212f8090b75ddc 2015-03-20 00:32:21
From: Jessica B. Hamrick <jhamrick@berkeley.edu>
Date: 2015-03-20 00:32:21
Subject: [PATCH] Also add a test to check that an error is raised on timeout by default

---

diff --git a/IPython/nbconvert/preprocessors/tests/test_execute.py b/IPython/nbconvert/preprocessors/tests/test_execute.py
index f6ea6d9..cb22ef5 100644
--- a/IPython/nbconvert/preprocessors/tests/test_execute.py
+++ b/IPython/nbconvert/preprocessors/tests/test_execute.py
@@ -11,12 +11,18 @@ import io
 import os
 import re
 
+try:
+    from queue import Empty  # Py 3
+except ImportError:
+    from Queue import Empty  # Py 2
+
 from IPython import nbformat
 
 from .base import PreprocessorTestsBase
 from ..execute import ExecutePreprocessor
 
 from IPython.nbconvert.filters import strip_ansi
+from nose.tools import assert_raises
 
 addr_pat = re.compile(r'0x[0-9a-f]{7,9}')
 
@@ -135,3 +141,9 @@ class TestExecute(PreprocessorTestsBase):
         self.assertEqual(output['output_type'], 'error')
         self.assertEqual(output['ename'], 'StdinNotImplementedError')
         self.assertEqual(output['evalue'], 'raw_input was called, but this frontend does not support input requests.')
+
+    def test_timeout(self):
+        """Check that an error is raised when a computation times out"""
+        current_dir = os.path.dirname(__file__)
+        filename = os.path.join(current_dir, 'files', 'Interrupt.ipynb')
+        assert_raises(Empty, self.run_notebook, filename, dict(timeout=1))