seems to me an engine would succumb to water damage before a transmission would. thinking about height of air intake and location of transmission breather.

The trans started to slip soon after going through the puddle?

And I think trans fluid with water in it shows tell-tale signs of water contamination.