Index: content/test/gpu/gpu_tests/gpu_test_base_unittest.py |
diff --git a/content/test/gpu/gpu_tests/gpu_test_base_unittest.py b/content/test/gpu/gpu_tests/gpu_test_base_unittest.py |
index f80ab48af28b1c0185a2438b298297c831a79592..a049688bca05539255fcbeea66c9c0cb4f5097b9 100644 |
--- a/content/test/gpu/gpu_tests/gpu_test_base_unittest.py |
+++ b/content/test/gpu/gpu_tests/gpu_test_base_unittest.py |
@@ -4,7 +4,7 @@ |
import os |
import unittest |
-from telemetry.core import discover |
+from telemetry.util import classes_util |
import gpu_test_base |
@@ -18,10 +18,8 @@ def _GetGpuDir(*subdirs): |
class NoOverridesTest(unittest.TestCase): |
def testValidatorBase(self): |
- all_validators = discover.DiscoverClasses( |
- _GetGpuDir('gpu_tests'), _GetGpuDir(), |
- gpu_test_base.ValidatorBase, |
- index_by_class_name=True).values() |
+ all_validators = classes_util.DiscoverClasses( |
+ _GetGpuDir('gpu_tests'), _GetGpuDir(), gpu_test_base.ValidatorBase) |
self.assertGreater(len(all_validators), 0) |
for validator in all_validators: |
self.assertEquals(gpu_test_base.ValidatorBase.ValidateAndMeasurePage, |
@@ -30,10 +28,8 @@ class NoOverridesTest(unittest.TestCase): |
% validator.__name__) |
def testPageBase(self): |
- all_pages = discover.DiscoverClasses( |
- _GetGpuDir(), _GetGpuDir(), |
- gpu_test_base.PageBase, |
- index_by_class_name=True).values() |
+ all_pages = classes_util.DiscoverClasses( |
+ _GetGpuDir(), _GetGpuDir(), gpu_test_base.PageBase) |
self.assertGreater(len(all_pages), 0) |
for page in all_pages: |
self.assertEquals(gpu_test_base.PageBase.RunNavigateSteps, |