fix issue43 - better tracebacks for unexpected exceptions in doctests

1 files changed, 15 insertions, 0 deletions

diff --git a/testing/test_doctest.py b/testing/test_doctest.py
index ed9cca9fd..67dc7fdbf 100644
--- a/testing/test_doctest.py
+++ b/testing/test_doctest.py
@@ -59,6 +59,21 @@ class TestDoctests:
             "*UNEXPECTED*ZeroDivision*",
         ])
 
+    def test_doctest_unex_importerror(self, testdir):
+        testdir.tmpdir.join("hello.py").write(py.code.Source("""
+            import asdalsdkjaslkdjasd
+        """))
+        p = testdir.maketxtfile("""
+            >>> import hello
+            >>>
+        """)
+        result = testdir.runpytest("--doctest-modules")
+        result.stdout.fnmatch_lines([
+            "*>>> import hello",
+            "*UNEXPECTED*ImportError*",
+            "*import asdals*",
+        ])
+
     def test_doctestmodule(self, testdir):
         p = testdir.makepyfile("""
             '''