[PATCH 1/2] tests: Don't use assertRaises as a context manager