Add a context argument to the setup and teardown functions.