tools/chrome_proxy/live_tests/pagesets/__init__.py

   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
   3 # found in the LICENSE file.
   4
   5 import inspect
   6 import os
   7 import sys
   8
   9 from telemetry.core import discover
  10 from telemetry import story
  11
  12
  13 # Import all submodules' StorySet classes.
  14 start_dir = os.path.dirname(os.path.abspath(__file__))
  15 top_level_dir = os.path.abspath(os.path.join(start_dir, os.pardir, os.pardir))
  16 base_class = story.StorySet
  17 for cls in discover.DiscoverClasses(
  18     start_dir, top_level_dir, base_class).values():
  19   setattr(sys.modules[__name__], cls.__name__, cls)
  20