Index: tools/telemetry/telemetry/core/util_unittest.py |
diff --git a/tools/telemetry/telemetry/core/util_unittest.py b/tools/telemetry/telemetry/core/util_unittest.py |
index 15a1aaff4a40c7d39a05bc38dc1c93996d0cc4de..a0b3de22ed5bb1ef8c22529fcd203a2d9f39bd6b 100644 |
--- a/tools/telemetry/telemetry/core/util_unittest.py |
+++ b/tools/telemetry/telemetry/core/util_unittest.py |
@@ -6,6 +6,7 @@ import shutil |
import tempfile |
import unittest |
+from telemetry.core import exceptions |
from telemetry.core import util |
@@ -18,7 +19,8 @@ class TestWait(unittest.TestCase): |
def testTimeout(self): |
def test(): |
return False |
- self.assertRaises(util.TimeoutException, lambda: util.WaitFor(test, 0.1)) |
+ self.assertRaises(exceptions.TimeoutException, |
+ lambda: util.WaitFor(test, 0.1)) |
def testCallable(self): |
"""Test methods and anonymous functions, functions are tested elsewhere.""" |