| 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."""
|
|
|