diff options
author | Ned Batchelder <ned@nedbatchelder.com> | 2010-03-13 20:20:24 -0500 |
---|---|---|
committer | Ned Batchelder <ned@nedbatchelder.com> | 2010-03-13 20:20:24 -0500 |
commit | 9f99a6e301a261b1f413292f0a9d9ed5bc44a0c7 (patch) | |
tree | a5c797e820490ced90b8d25ef52cafe7af5fe5c9 /test/coveragetest.py | |
parent | afd257b85732ab6b10ec88dd48ddd5688dde3834 (diff) | |
download | python-coveragepy-git-9f99a6e301a261b1f413292f0a9d9ed5bc44a0c7.tar.gz |
When fiddling with PYTHONPATH, be prepared for it not to exist.
Diffstat (limited to 'test/coveragetest.py')
-rw-r--r-- | test/coveragetest.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/test/coveragetest.py b/test/coveragetest.py index 54b4bd43..c82891e4 100644 --- a/test/coveragetest.py +++ b/test/coveragetest.py @@ -92,7 +92,7 @@ class CoverageTest(TestCase): if name in self.environ_undos: return self.environ_undos[name] else: - return os.environ[name] + return os.environ.get(name) def undo_environ(self): """Undo all the changes made by `set_environ`.""" @@ -327,6 +327,8 @@ class CoverageTest(TestCase): pypath = self.original_environ('PYTHONPATH') if pypath: pypath += os.pathsep + else: + pypath = "" pypath += testmods + os.pathsep + zipfile self.set_environ('PYTHONPATH', pypath) |