<s> import contextlib <EOL> import os <EOL> @ contextlib . contextmanager <EOL> def working_dir ( new_dir ) : <EOL> before = os . getcwd ( ) <EOL> os . chdir ( new_dir ) <EOL> yield <EOL> os . chdir ( before ) </s>
