They shouldn’t be asking you to code anything related to their actual work — it should be a toy assignment.
I think it all comes down to personal pref, but the data seems to show that whiteboard interviews are good to filter middle of the road talent, but top/bottom can look the same.