Correct formatting for Python 2.6 compat
Change test to test context (context manager)