style changes for contextmanager test

hah was going to do this.. I personally don't like it but I guess it is more pythonic :)