Shortened test time for utils/wait.py

Reduced time for test to run