Can babies wear diapers in the water?

So I know this might be a dumb question but I have no idea. I got my son a pool and was wondering if he can go in there with his diaper and what does he want a shorts bathing suit? Please help ftm