Update swim_tests.py

Changed import reference