[Python-checkins] r63497 - python/trunk/Lib/test/test_pydoc.py

[georg.brandl]

Author: georg.brandl
Date: Tue May 20 10:10:03 2008
New Revision: 63497

Log:
Use inspect.getabsfile() to get the documented module's filename.


Modified:
   python/trunk/Lib/test/test_pydoc.py

Modified: python/trunk/Lib/test/test_pydoc.py
==============================================================================
--- python/trunk/Lib/test/test_pydoc.py	(original)
+++ python/trunk/Lib/test/test_pydoc.py	Tue May 20 10:10:03 2008
@@ -3,6 +3,7 @@
 import subprocess
 import re
 import pydoc
+import inspect
 import unittest
 import test.test_support
 
@@ -196,17 +197,12 @@
                                  tofile='got')
     print '\n' + ''.join(diffs)
 
-def get_mod_file(module):
-    """Returns the name of the .py file if the module is compiled"""
-    mod_file = module.__file__
-    return mod_file if mod_file.endswith("py") else mod_file[:-3] + "py"
-
 
 class PyDocDocTest(unittest.TestCase):
 
     def test_html_doc(self):
         result = get_pydoc_html(pydoc_mod)
-        mod_file = get_mod_file(pydoc_mod)
+        mod_file = inspect.getabsfile(pydoc_mod)
         expected_html = expected_html_pattern % (mod_file, mod_file)
         if result != expected_html:
             print_diffs(expected_html, result)
@@ -214,7 +210,7 @@
 
     def test_text_doc(self):
         result = get_pydoc_text(pydoc_mod)
-        expected_text = expected_text_pattern % get_mod_file(pydoc_mod)
+        expected_text = expected_text_pattern % inspect.getabsfile(pydoc_mod)
         if result != expected_text:
             print_diffs(expected_text, result)
             self.fail("outputs are not equal, see diff above")