Index: tools/telemetry/telemetry/internal/story_runner_unittest.py |
diff --git a/tools/telemetry/telemetry/internal/story_runner_unittest.py b/tools/telemetry/telemetry/internal/story_runner_unittest.py |
index 65959c76279993ba25746024c109d046028d2074..ac26acf57fda61ec29c7436331149056224df904 100644 |
--- a/tools/telemetry/telemetry/internal/story_runner_unittest.py |
+++ b/tools/telemetry/telemetry/internal/story_runner_unittest.py |
@@ -175,18 +175,18 @@ class StoryRunnerTest(unittest.TestCase): |
# UserStorySet's are only allowed to have one SharedState. |
us = SetupUserStorySet(False, foo_states) |
story_groups = ( |
- story_runner.StoriesGroupedByStateClass( |
+ story_runner.StoriesByStateClass( |
us, False)) |
self.assertEqual(len(story_groups), 1) |
us = SetupUserStorySet(False, mixed_states) |
self.assertRaises( |
ValueError, |
- story_runner.StoriesGroupedByStateClass, |
+ story_runner.StoriesByStateClass, |
us, False) |
# BaseUserStorySets are allowed to have multiple SharedStates. |
bus = SetupUserStorySet(True, mixed_states) |
story_groups = ( |
- story_runner.StoriesGroupedByStateClass( |
+ story_runner.StoriesByStateClass( |
bus, True)) |
self.assertEqual(len(story_groups), 3) |
self.assertEqual(story_groups[0].shared_state_class, |