It would be fine as long as you use fused disconnects at the taps.
I don't see much specific benefit to keeping them separate.
But I have no idea which way will work out better as far as costs.
By the way some power companies have rules regarding these units, it may come back to haunt the customer if their new water heater makes the neighbors lights flicker.
