33 lines
823 B
Python
33 lines
823 B
Python
from . import util as source_util
|
|
|
|
from importlib import machinery
|
|
import imp
|
|
import unittest
|
|
|
|
|
|
class PathHookTest(unittest.TestCase):
|
|
|
|
"""Test the path hook for source."""
|
|
|
|
def path_hook(self):
|
|
return machinery.FileFinder.path_hook((machinery.SourceFileLoader,
|
|
machinery.SOURCE_SUFFIXES, True))
|
|
|
|
def test_success(self):
|
|
with source_util.create_modules('dummy') as mapping:
|
|
self.assertTrue(hasattr(self.path_hook()(mapping['.root']),
|
|
'find_module'))
|
|
|
|
def test_empty_string(self):
|
|
# The empty string represents the cwd.
|
|
self.assertTrue(hasattr(self.path_hook()(''), 'find_module'))
|
|
|
|
|
|
def test_main():
|
|
from test.support import run_unittest
|
|
run_unittest(PathHookTest)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
test_main()
|