In my campaign:
Heavy armor/heavy encumbrance = you have -20ft speed, you can't stay afloat but you can walk on the bottom or climb walls
Medium armor/encumbered = you have -10ft speed, you can swim with disadvantage
Light armor/no encumbrance = you can swim normally