diff options
author | Ned Batchelder <ned@nedbatchelder.com> | 2015-11-22 10:03:13 -0500 |
---|---|---|
committer | Ned Batchelder <ned@nedbatchelder.com> | 2015-11-22 10:03:13 -0500 |
commit | 3406ff2500371bdb307d6c94aa104403d2b18871 (patch) | |
tree | 065ecb2a475092c0d8a81ce1f7b0567009e823a4 /coverage/backward.py | |
parent | fa70cc894ad8b17e915a96c84c77b7fc38b92f0b (diff) | |
download | python-coveragepy-git-3406ff2500371bdb307d6c94aa104403d2b18871.tar.gz |
Extend import_local_file so I can use a file in another directory.
Diffstat (limited to 'coverage/backward.py')
-rw-r--r-- | coverage/backward.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/coverage/backward.py b/coverage/backward.py index 81ca342f..4fc72215 100644 --- a/coverage/backward.py +++ b/coverage/backward.py @@ -151,11 +151,12 @@ except AttributeError: PYC_MAGIC_NUMBER = imp.get_magic() -def import_local_file(modname): +def import_local_file(modname, modfile=None): """Import a local file as a module. Opens a file in the current directory named `modname`.py, imports it - as `modname`, and returns the module object. + as `modname`, and returns the module object. `modfile` is the file to + import if it isn't in the current directory. """ try: @@ -163,7 +164,8 @@ def import_local_file(modname): except ImportError: SourceFileLoader = None - modfile = modname + '.py' + if modfile is None: + modfile = modname + '.py' if SourceFileLoader: mod = SourceFileLoader(modname, modfile).load_module() else: |