Yeah, BPs swim fine. My hatchlings actually sometimes belly flop straight into their water bowl. They figure out swimming much faster than crawling. Though they scare me because I don't know if they understand they can't BREATHE underwater.
I generally only swim my bigger ball pythons when somebody's been rolling in a poo pile. Some of them kinda like it, and will hang out in the water blowing bubbles. Some of them really hate it, and will shoot out of the tub before they're even completely wet. And water temperature is a big deal---they really only like room temperature water, or just a little warmer. 75-80 degrees.