summaryrefslogtreecommitdiffstats
path: root/Lib/importlib/test/import_/util.py
blob: 6ab06518912c69aef133e61973dd1112e5386c67 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import functools
import importlib._bootstrap


using___import__ = False


def import_(*args, **kwargs):
    """Delegate to allow for injecting different implementations of import."""
    if using___import__:
        return __import__(*args, **kwargs)
    #return importlib.Import()(*args, **kwargs)
    return importlib._bootstrap._import(*args, **kwargs)


def importlib_only(fxn):
    """Decorator to mark which tests are not supported by the current
    implementation of __import__()."""
    def inner(*args, **kwargs):
        if using___import__:
            return
        else:
            return fxn(*args, **kwargs)
    functools.update_wrapper(inner, fxn)
    return inner


def mock_path_hook(*entries, importer):
    """A mock sys.path_hooks entry."""
    def hook(entry):
        if entry not in entries:
            raise ImportError
        return importer
    return hook