Dont use assertRaises as context manager

Breaks on 2.6.