TST: Clean up tests with assertRaises context manager.