mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-17 21:10:43 +00:00
36 lines
No EOL
963 B
Python
36 lines
No EOL
963 B
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
|
|
import sys
|
|
import random
|
|
import unittest
|
|
|
|
if __name__ == '__main__':
|
|
sys.path.append(sys.argv[4])
|
|
sys.path.append(sys.argv[5])
|
|
|
|
import function_tests
|
|
|
|
loader = unittest.TestLoader()
|
|
|
|
test_classes = [function_tests.HttpJsonPayloadTests, function_tests.HttpProtobufPayloadTests, function_tests.HttpEndpointTests, function_tests.GRPCTests]
|
|
|
|
test_suites = []
|
|
for tests in test_classes:
|
|
tests.server_app_path = sys.argv[1]
|
|
tests.model_path = sys.argv[2]
|
|
tests.test_data_path = sys.argv[3]
|
|
tests.server_port = random.randint(30000, 50000)
|
|
|
|
test_suites.append(loader.loadTestsFromTestCase(tests))
|
|
|
|
suites = unittest.TestSuite(test_suites)
|
|
runner = unittest.TextTestRunner(verbosity=2)
|
|
|
|
results = runner.run(suites)
|
|
|
|
if results.wasSuccessful():
|
|
exit(0)
|
|
else:
|
|
exit(1)
|
|
|