Index: telemetry/telemetry/benchmark_unittest.py |
diff --git a/telemetry/telemetry/benchmark_unittest.py b/telemetry/telemetry/benchmark_unittest.py |
index cb23c6f18eaf4508bda61e026940755d8728ae8c..1710b1f9f32c7c03f361803d556417fb9882bbe8 100644 |
--- a/telemetry/telemetry/benchmark_unittest.py |
+++ b/telemetry/telemetry/benchmark_unittest.py |
@@ -159,3 +159,9 @@ class BenchmarkTest(unittest.TestCase): |
story_runner.Run = original_run_fn |
self.assertTrue(validPredicate[0]) |
+ |
+ def testBenchmarkExpectations(self): |
+ b = TestBenchmark(story_module.Story( |
+ shared_state_class=shared_page_state.SharedPageState)) |
+ self.assertIsInstance( |
+ b.GetExpectations(), story_module.expectations.StoryExpectations) |