1#!/usr/bin/env python3 2# Copyright (C) 2020 The Android Open Source Project 3# 4# Licensed under the Apache License, Version 2.0 (the "License"); 5# you may not use this file except in compliance with the License. 6# You may obtain a copy of the License at 7# 8# http://www.apache.org/licenses/LICENSE-2.0 9# 10# Unless required by applicable law or agreed to in writing, software 11# distributed under the License is distributed on an "AS IS" BASIS, 12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13# See the License for the specific language governing permissions and 14# limitations under the License. 15 16import argparse 17import os 18import sys 19import unittest 20 21from test import api_unittest 22from test import api_integrationtest 23from test import resolver_unittest 24from test import stdlib_unittest 25 26ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) 27 28 29def main(): 30 try: 31 import numpy 32 import pandas 33 except ModuleNotFoundError: 34 print('Cannot proceed. Please `pip3 install pandas numpy`', file=sys.stderr) 35 return 1 36 37 # Set paths to trace_processor_shell and root directory as environment 38 # variables 39 parser = argparse.ArgumentParser() 40 parser.add_argument("shell", type=str) 41 os.environ["SHELL_PATH"] = parser.parse_args().shell 42 os.environ["ROOT_DIR"] = ROOT_DIR 43 44 loader = unittest.TestLoader() 45 suite = unittest.TestSuite() 46 47 # Add all relevant tests to test suite 48 suite.addTests(loader.loadTestsFromModule(api_unittest)) 49 suite.addTests(loader.loadTestsFromModule(resolver_unittest)) 50 suite.addTests(loader.loadTestsFromModule(api_integrationtest)) 51 suite.addTests(loader.loadTestsFromModule(stdlib_unittest)) 52 53 # Initialise runner to run all tests in suite 54 runner = unittest.TextTestRunner(verbosity=3) 55 result = runner.run(suite) 56 57 return 0 if result.wasSuccessful() else 1 58 59 60if __name__ == '__main__': 61 sys.exit(main()) 62