| OLD | NEW |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 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 import os | 4 import os |
| 5 import sys | 5 import sys |
| 6 | 6 |
| 7 from telemetry.core import discover | 7 from telemetry.core import discover |
| 8 from telemetry.story import story_set | 8 from telemetry.story import story_set |
| 9 | 9 |
| 10 | 10 from page_sets.gpu_process_tests import GpuProcessTestsStorySet |
| 11 # Import all submodules' StorySet classes. | 11 from page_sets.gpu_rasterization_tests import GpuRasterizationTestsStorySet |
| 12 start_dir = os.path.dirname(os.path.abspath(__file__)) | 12 from page_sets.memory_tests import MemoryTestsStorySet |
| 13 top_level_dir = os.path.dirname(start_dir) | 13 from page_sets.pixel_tests import PixelTestsStorySet |
| 14 base_class = story_set.StorySet | |
| 15 for cls in discover.DiscoverClasses( | |
| 16 start_dir, top_level_dir, base_class).values(): | |
| 17 setattr(sys.modules[__name__], cls.__name__, cls) | |
| OLD | NEW |