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__))) 27sys.path.append(os.path.join(ROOT_DIR)) 28 29 30def main(): 31 try: 32 import numpy 33 import pandas 34 except ModuleNotFoundError: 35 print('Cannot proceed. Please `pip3 install pandas numpy`', file=sys.stderr) 36 return 1 37 38 # Set paths to trace_processor_shell and root directory as environment 39 # variables 40 parser = argparse.ArgumentParser() 41 parser.add_argument("shell", type=str) 42 os.environ["SHELL_PATH"] = parser.parse_args().shell 43 os.environ["ROOT_DIR"] = ROOT_DIR 44 45 loader = unittest.TestLoader() 46 suite = unittest.TestSuite() 47 48 # Add all relevant tests to test suite 49 suite.addTests(loader.loadTestsFromModule(api_unittest)) 50 suite.addTests(loader.loadTestsFromModule(resolver_unittest)) 51 suite.addTests(loader.loadTestsFromModule(api_integrationtest)) 52 suite.addTests(loader.loadTestsFromModule(stdlib_unittest)) 53 54 # Initialise runner to run all tests in suite 55 runner = unittest.TextTestRunner(verbosity=3) 56 result = runner.run(suite) 57 58 return 0 if result.wasSuccessful() else 1 59 60 61if __name__ == '__main__': 62 sys.exit(main()) 63