use wrapt for the TestContext decorator

I think this got missed in #36.