Use the existing all_warnings context manager