def test__exit(self): "Test TemporaryDirectory __exit__() method" tdir = TemporaryDirectory() trace = """ Called tempfile.mkdtemp('', 'tmp', None) Called shutil.rmtree('/fake/tmp/dir')""" tdir.__exit__(None, None, None) assert_same_trace(self.tt, trace)
def test_change_cwd(self): "Test TemporaryDirectory changing the cwd" expected = "/fake/tmp/dir" trace = """ Called tempfile.mkdtemp('', 'tmp', None) Called os.getcwd() Called os.chdir('/fake/tmp/dir') Called os.chdir('/old/cur/dir') Called shutil.rmtree('/fake/tmp/dir')""" tdir = TemporaryDirectory(change=True) tdir.__enter__() self.assertEqual(self.cwd, expected) self.assertEqual(tdir.old_wd, '/old/cur/dir') tdir.__exit__(None, None, None) self.assertEqual(self.cwd, '/old/cur/dir') assert_same_trace(self.tt, trace)