fix issue43 - better tracebacks for unexpected exceptions in doctests

2011-05-28 14:38:15 +02:00
parent 8dc4e732f0
commit 410438f187
3 changed files with 18 additions and 1 deletions
@@ -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("""
            '''