systrace/catapult/devil/bin/run_py_devicetests

#!/usr/bin/env vpython
# Copyright 2016 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import logging
import os
import sys

_CATAPULT_PATH = os.path.abspath(
    os.path.join(os.path.dirname(__file__), '..', '..'))
_DEVIL_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
_TYP_PATH = os.path.abspath(os.path.join(_CATAPULT_PATH, 'third_party', 'typ'))

sys.path.append(_TYP_PATH)
import typ

sys.path.append(_DEVIL_PATH)
from devil.android import device_test_case


def _SetUpLogging():
  parsed_args = typ.arg_parser.ArgumentParser().parse_args(args=sys.argv[1:])
  verbosity = parsed_args.verbose
  level = None
  if verbosity == 0:
    level = logging.WARNING
  elif verbosity == 1:
    level = logging.INFO
  elif verbosity >= 2:
    level = logging.DEBUG
  else:
    raise RuntimeError(
        'Logging verbosity of {} is not allowed.'.format(verbosity))
  logging.basicConfig(level=level)


def main():
  _SetUpLogging()
  runner = typ.Runner()
  runner.setup_fn = device_test_case.PrepareDevices
  return runner.main(
      coverage_source=[_DEVIL_PATH],
      jobs=1,
      suffixes=['*_devicetest.py'],
      top_level_dir=_DEVIL_PATH)


if __name__ == '__main__':
  sys.exit(main())