Use context manager form of assertRaises