| 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,
|
|
|