def __enter__(self):
self.tempdir = tempfile.mkdtemp()
self.location = os.path.abspath(os.curdir)
os.chdir(self.tempdir)
wrapper = Wrapper()
wrapper.setup_run(setup_docs=False)
return wrapper
def test_wrapper_register():
with tempdir():
doc = Doc("abc.txt")
wrapper = Wrapper()
wrapper.setup_run()
wrapper.register(doc)
assert doc in wrapper.registered
