Avoid KeyError when preserving an environment key that does not exist

2014-01-04 17:44:04 +00:00 · 2014-01-04 17:44:04 +00:00 · 3fadf28751
parent d5aaf8af0e
commit 3fadf28751
1 changed files with 1 additions and 1 deletions
--- a/2
+++ b/2
@ -35,7 +35,7 @@ class ConfigureTests:
 			return f
 	class PreservedEnvironment:
 		def __init__(self,env,keys):
-			self.flags = {k: env[k][:] for k in keys}
+			self.flags = {k: env.get(k, [])[:] for k in keys}
 		def restore(self,env):
 			env.Replace(**self.flags)
 		def __getitem__(self,key):