You can switch MX records but not necessarily your mail storage. You need IMAP for that, and IMAP with Proton currently requires jumping through some hoops and it may be discontinued in the near future.

They’ve never given any guarantees regarding IMAP and they actually seem to consider it a negative so that remains a dubious point with me.