[Telemetry] Move from telemetry WaitFor to py_utils WaitFor

telemetry/telemetry/internal/actions/load_media_unittest.py

Index: telemetry/telemetry/internal/actions/load_media_unittest.py
diff --git a/telemetry/telemetry/internal/actions/load_media_unittest.py b/telemetry/telemetry/internal/actions/load_media_unittest.py
index 34e24b98d480568a0d2693e5ceda367f16e6a609..ea7dd517af02981edb6a41a9220004eacc39ca09 100644
--- a/telemetry/telemetry/internal/actions/load_media_unittest.py
+++ b/telemetry/telemetry/internal/actions/load_media_unittest.py
@@ -2,11 +2,12 @@
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
-from telemetry.core import exceptions
 from telemetry import decorators
 from telemetry.internal.actions.load_media import LoadMediaAction
 from telemetry.testing import tab_test_case
 
+import py_utils
+
 
 class LoadMediaActionTest(tab_test_case.TabTestCase):
 
@@ -61,4 +62,4 @@ class LoadMediaActionTest(tab_test_case.TabTestCase):
     action = LoadMediaAction(selector='#video_1', timeout_in_seconds=0.1,
                              event_to_await='a_nonexistent_event')
     action.WillRunAction(self._tab)
-    self.assertRaises(exceptions.TimeoutException, action.RunAction, self._tab)
+    self.assertRaises(py_utils.TimeoutException, action.RunAction, self._tab)