test/py/test.py

   1 #!/usr/bin/env python3
   2 # SPDX-License-Identifier: GPL-2.0
   3
   4 # Copyright (c) 2015 Stephen Warren
   5 # Copyright (c) 2015-2016, NVIDIA CORPORATION. All rights reserved.
   6
   7 # Wrapper script to invoke pytest with the directory name that contains the
   8 # U-Boot tests.
   9
  10 import os
  11 import os.path
  12 import sys
  13 import pytest
  14
  15 if __name__ == '__main__':
  16     # argv; py.test test_directory_name user-supplied-arguments
  17     args = [os.path.dirname(__file__) + '/tests']
  18     args.extend(sys.argv)
  19
  20     # Use short format by default
  21     if not [arg for arg in args if '--tb=' in arg]:
  22         args.append('--tb=short')
  23
  24     sys.exit(pytest.main(args))