Although issue #19 was a doctest flaw, it's good to have a test to be sure we're still working well with doctests.

1 files changed, 44 insertions, 0 deletions

diff --git a/test/test_oddball.py b/test/test_oddball.py
index 57d2fdbd..0376f65a 100644
--- a/test/test_oddball.py
+++ b/test/test_oddball.py
@@ -240,3 +240,47 @@ class ExceptionTest(CoverageTest):
                     continue
                 clean_lines[os.path.basename(f)] = llist
             self.assertEqual(clean_lines, lines_expected)
+
+
+if sys.hexversion > 0x02050000:
+    class DoctestTest(CoverageTest):
+        """Tests invoked with doctest should measure properly."""
+    
+        def setUp(self):
+            super(DoctestTest, self).setUp()
+            
+            # Oh, the irony!  This test case exists because Python 2.4's
+            # doctest module doesn't play well with coverage.  But nose fixes
+            # the problem by monkeypatching doctest.  I want to undo the
+            # monkeypatch to be sure I'm getting the doctest module that users
+            # of coverage will get.  Deleting the imported module here is
+            # enough: when the test imports doctest again, it will get a fresh
+            # copy without the monkeypatch.
+            del sys.modules['doctest']
+    
+        def testDoctest(self):
+            self.check_coverage('''\
+                def return_arg_or_void(arg):
+                    """If <arg> is None, return "Void"; otherwise return <arg>
+                    
+                    >>> return_arg_or_void(None)
+                    'Void'
+                    >>> return_arg_or_void("arg")
+                    'arg'
+                    >>> return_arg_or_void("None")
+                    'None'
+                    """
+                    if arg is None:
+                        return "Void"
+                    else:
+                        return arg
+                
+                import doctest, sys
+                doctest.testmod(sys.modules[__name__])  # we're not __main__ :(
+                ''',
+                [1,11,12,14,16,17], "")
+
+
+if __name__ == '__main__':
+    import unittest
+    unittest.main()