OLD | NEW |
1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import inspect | 5 import inspect |
6 import os | 6 import os |
7 import sys | 7 import sys |
8 | 8 |
9 from telemetry.core import discover | 9 from telemetry.core import discover |
10 from telemetry import story | 10 from telemetry import story |
| 11 from telemetry.story import expectations |
11 | 12 |
12 | 13 |
13 # Import all submodules' PageSet classes. | 14 # Import all submodules' PageSet classes. |
14 start_dir = os.path.dirname(os.path.abspath(__file__)) | 15 start_dir = os.path.dirname(os.path.abspath(__file__)) |
15 top_level_dir = os.path.dirname(start_dir) | 16 top_level_dir = os.path.dirname(start_dir) |
16 base_classes = [story.StorySet, story.expectations.StoryExpectations] | 17 base_classes = [story.StorySet, expectations.StoryExpectations] |
17 | 18 |
18 for base_class in base_classes: | 19 for base_class in base_classes: |
19 for cls in discover.DiscoverClasses( | 20 for cls in discover.DiscoverClasses( |
20 start_dir, top_level_dir, base_class).values(): | 21 start_dir, top_level_dir, base_class).values(): |
21 setattr(sys.modules[__name__], cls.__name__, cls) | 22 setattr(sys.modules[__name__], cls.__name__, cls) |
OLD | NEW |