Index: tools/telemetry/telemetry/benchmark_runner_unittest.py |
diff --git a/tools/telemetry/telemetry/benchmark_runner_unittest.py b/tools/telemetry/telemetry/benchmark_runner_unittest.py |
index 68fc80606219f7757d898c4db53b357a07c89b2a..e28dcc14dcd39351bd4c5f855dac6f5ee063975d 100644 |
--- a/tools/telemetry/telemetry/benchmark_runner_unittest.py |
+++ b/tools/telemetry/telemetry/benchmark_runner_unittest.py |
@@ -27,6 +27,11 @@ class BenchmarkBar(benchmark.Benchmark): |
def Name(cls): |
return 'BarBenchmarkkkkk' |
+class UnusualBenchmark(benchmark.Benchmark): |
+ @classmethod |
+ def Name(cls): |
+ return 'I have a very unusual name' |
+ |
class BenchmarkRunnerUnittest(unittest.TestCase): |
def setUp(self): |
@@ -65,3 +70,20 @@ class BenchmarkRunnerUnittest(unittest.TestCase): |
benchmark_runner.PrintBenchmarkList( |
[BenchmarkFoo, BenchmarkBar], self._mock_possible_browser, self._stream) |
self.assertEquals(expected_printed_stream, self._stream.output_data) |
+ |
+ def testGetMostLikelyMatchedBenchmarks(self): |
+ all_benchmarks = [BenchmarkFoo, BenchmarkBar, UnusualBenchmark] |
+ self.assertEquals( |
+ [BenchmarkFoo, BenchmarkBar], |
+ benchmark_runner.GetMostLikelyMatchedBenchmarks( |
+ all_benchmarks, 'BenchmarkFooz')) |
+ |
+ self.assertEquals( |
+ [BenchmarkBar, BenchmarkFoo], |
+ benchmark_runner.GetMostLikelyMatchedBenchmarks( |
+ all_benchmarks, 'BarBenchmark')) |
+ |
+ self.assertEquals( |
+ [UnusualBenchmark], |
+ benchmark_runner.GetMostLikelyMatchedBenchmarks( |
+ all_benchmarks, 'unusual')) |