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
10import os
11import os.path
12import sys
13import pytest
14from pkg_resources import load_entry_point
15
16if __name__ == '__main__':
17    # argv; py.test test_directory_name user-supplied-arguments
18    args = [os.path.dirname(__file__) + '/tests']
19    args.extend(sys.argv)
20
21    # Use short format by default
22    if not [arg for arg in args if '--tb=' in arg]:
23        args.append('--tb=short')
24
25    sys.exit(pytest.main(args))
26