Index: tools/perf/benchmarks/benchmark_unittest.py |
diff --git a/tools/perf/benchmarks/benchmark_unittest.py b/tools/perf/benchmarks/benchmark_unittest.py |
index d5d039c547f41a0171a0dd143774bd18bdf79a06..9fe848d1a1857b9409b78a6951bd76c785915aaa 100644 |
--- a/tools/perf/benchmarks/benchmark_unittest.py |
+++ b/tools/perf/benchmarks/benchmark_unittest.py |
@@ -11,9 +11,9 @@ |
from core import perf_benchmark |
from telemetry import benchmark as benchmark_module |
+from telemetry.core import discover |
from telemetry.internal.browser import browser_options |
from telemetry.testing import progress_reporter |
-from telemetry.util import classes_util |
def _GetPerfDir(*subdirs): |
@@ -22,8 +22,9 @@ |
def _GetAllPerfBenchmarks(): |
- return classes_util.DiscoverClasses( |
- _GetPerfDir('benchmarks'), _GetPerfDir(), benchmark_module.Benchmark) |
+ return discover.DiscoverClasses( |
+ _GetPerfDir('benchmarks'), _GetPerfDir(), benchmark_module.Benchmark, |
+ index_by_class_name=True).values() |
def _BenchmarkOptionsTestGenerator(benchmark): |
def testBenchmarkOptions(self): # pylint: disable=W0613 |