diff --git a/tests/module_finder_test.py b/tests/module_finder_test.py index 9cc6c58a..ef58fa06 100644 --- a/tests/module_finder_test.py +++ b/tests/module_finder_test.py @@ -1,5 +1,6 @@ - +import inspect import unittest + import mitogen.master import testlib @@ -74,7 +75,7 @@ class GetModuleViaPkgutilTest(testlib.TestCase): path, src, is_pkg = self.call('module_finder_testmod.regular_mod') self.assertEquals(path, testlib.data_path('module_finder_testmod/regular_mod.py')) - self.assertEquals(src, file(regular_mod.__file__).read()) + self.assertEquals(src, inspect.getsource(regular_mod)) self.assertFalse(is_pkg)