Update swim_tests.py

Corrected swim tests issue