Context manager's best practice for set-up then run function

Question

I have a project where I need to set up many folder, run a simulation, then tear down other folders. I would like to use context manager to do that, but I am not sure which