A west Australian police officer caught a suspect mid-swim after the man jumped into the water during a police chase.